litert-community
/

inception_v3

Image Classification

computer-vision

Eval Results (legacy)

Model card Files Files and versions

fix_inception_v3

#2

by sourcelite - opened Apr 2

base: refs/heads/main

←

from: refs/pr/2

Discussion Files changed

Files changed (1) hide show

README.md +4 -1

README.md CHANGED Viewed

@@ -73,7 +73,10 @@ def preprocess(img: Image.Image) -> np.ndarray:
     x = (x - np.array([0.485, 0.456, 0.406], dtype=np.float32)) / np.array(
         [0.229, 0.224, 0.225], dtype=np.float32
     )
-    return np.transpose(x, (2, 0, 1))
 def main():
     ap = argparse.ArgumentParser()

     x = (x - np.array([0.485, 0.456, 0.406], dtype=np.float32)) / np.array(
         [0.229, 0.224, 0.225], dtype=np.float32
     )
+    # Expand dimensions to create NHWC 4D tensor: (1, 299, 299, 3)
+    x = np.expand_dims(x, axis=0)
+    return x
 def main():
     ap = argparse.ArgumentParser()