fix: correct image normalization in RAM preprocessing (#657)

CVHub520 · Oct 15, 2024 · 0b07dc4 · 0b07dc4
1 parent 41799b5
commit 0b07dc4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/anylabeling/services/auto_labeling/__base__/ram.py b/anylabeling/services/auto_labeling/__base__/ram.py
@@ -67,6 +67,7 @@ def preprocess(self, input_image, input_shape):
         """
         h, w = input_shape
         image = cv2.resize(input_image, (w, h))
+        image /= 255.0
         mean = np.array([0.485, 0.456, 0.406])
         std = np.array([0.229, 0.224, 0.225])
         image = (image - mean) / std

diff --git a/anylabeling/services/auto_labeling/yolov5_ram.py b/anylabeling/services/auto_labeling/yolov5_ram.py
@@ -113,6 +113,7 @@ def ram_preprocess(self, input_image):
         """
         h, w = self.ram_input_shape
         image = cv2.resize(input_image, (w, h))
+        image /= 255.0
         mean = np.array([0.485, 0.456, 0.406])
         std = np.array([0.229, 0.224, 0.225])
         image = (image - mean) / std