correct values

2025-04-21 03:44:08 +00:00 · 2025-04-21 03:44:08 +00:00 · e0852ccbf1
commit e0852ccbf1
parent 1456280e28
1 changed files with 2 additions and 1 deletions
--- a/inference_test.py
+++ b/inference_test.py
@ -113,7 +113,8 @@ def preprocess_image(image_path):
    mat_in = ncnn.Mat.from_pixels_resize(
        img, ncnn.Mat.PixelType.PIXEL_RGB, img.shape[1], img.shape[0], 224, 224
    )
-    mean_vals, norm_vals = [0.5] * 4, [1 / 255.0] * 4
+    mean_vals = [0.5, 0.5, 0.5] * 2
+    norm_vals = [1/255, 1/255, 1/255] * 2
    mat_in.substract_mean_normalize(mean_vals, norm_vals)
    return mat_in