Main fix classification (#15)

AXERA-TECH · Dec 10, 2024 · 638250d · 638250d
1 parent 57a6028
commit 638250d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/classification.py b/examples/classification.py
@@ -46,7 +46,7 @@ def preprocess_image(image_path, target_size=(256, 256), crop_size=(224, 224)):
     img_array = np.array(img).astype("uint8")
     # Transpose to (1, C, H, W)
     # img_array = np.transpose(img_array, (2, 0, 1))
-    # img_array = np.expand_dims(img_array, axis=0)  # Add batch dimension
+    img_array = np.expand_dims(img_array, axis=0)  # Add batch dimension
     return img_array