hsv

Author: armarpc

mmseg/datasets/pipelines/hubmap_transforms.py

@@ -22,7 +22,9 @@ def apply(self, image, hue_shift=0, sat_shift=0, val_shift=0, **params):
         if not A.is_rgb_image(image) and not A.is_grayscale_image(image):
             raise TypeError("HueSaturationValue transformation expects 1-channel or 3-channel images.")
 
-        image = (image*255).astype(np.uint8)
+        if image.dtype != np.uint8:
+            raise TypeError("image must be uint8")
+
         hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
 
         h = hsv[:, :, 0].astype(np.float32)  # hue
@@ -36,7 +38,7 @@ def apply(self, image, hue_shift=0, sat_shift=0, val_shift=0, **params):
         hsv[:, :, 1] = np.clip(s,0,255).astype(np.uint8)
         hsv[:, :, 2] = np.clip(v,0,255).astype(np.uint8)
         image = cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR)
-        image = image.astype(np.float32)/255
+        
         return image
 
     def get_params(self):