Tidying comments

carlyrichmond · Apr 25, 2024 · 214a486 · 214a486
1 parent 0195b97
commit 214a486
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/model-classification-app/scripts/transfer-learning.js b/model-classification-app/scripts/transfer-learning.js
@@ -41,6 +41,8 @@ async function run() {
 
   // Get features from MobileNet
   const mobileNetFeatures = images.map((image) => {
+    // Normalize the image data. Image data is always in the range of 0 to 255, 
+    // so you can simply divide resizedTensorFrame by 255 to ensure all values are between 0 and 1 instead as MobileNet expects
     const normalizedImageTensor = image.div(255);
     return mobileNetModel.predict(normalizedImageTensor.expandDims()).squeeze();
   });