Fix

SlinkoIgor · Mar 17, 2023 · c5da630 · c5da630
1 parent d0251cc
commit c5da630
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/common/data_utils.py b/common/data_utils.py
@@ -1134,11 +1134,11 @@ def preprocess_image(image, model_input_shape, uint8_mode=False, input_scale=Non
     """
     #resized_image = cv2.resize(image, model_input_shape[::-1], cv2.INTER_AREA)
     resized_image = letterbox_resize(image, model_input_shape[::-1])
-    image_data = normalize_image(image_data)
+    resized_image = normalize_image(resized_image)
 
     output_type = 'float32'
     if uint8_mode:
-        image_data = image_data / input_scale + input_zero_point
+        resized_image = resized_image / input_scale + input_zero_point
         output_type = 'uint8'
 
     image_data = np.asarray(resized_image).astype(output_type)