heavier test

2025-04-19 05:42:59 +00:00 · 2025-04-19 05:42:59 +00:00 · 1e05e1f65b
commit 1e05e1f65b
parent f956324616
1 changed files with 1 additions and 1 deletions
--- a/inference_test.py
+++ b/inference_test.py
@ -85,7 +85,7 @@ 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] * 2, [1 / 255.0] * 2
+    mean_vals, norm_vals = [0.5] * 4, [1 / 255.0] * 4
    mat_in.substract_mean_normalize(mean_vals, norm_vals)
    return mat_in