a

Author: armarpc

mmseg/datasets/pipelines/hubmap_transforms.py

@@ -2,6 +2,7 @@
 import numpy as np
 import albumentations as A
 from albumentations.pytorch import ToTensorV2
+import cv2
 
 def aug_mega_hardcore(p=.95):
     return A.Compose([
@@ -56,6 +57,11 @@ def __call__(self, results):
         transformed_image = transformed['image']
         transformed_masks = transformed['masks']
 
+        transformed_image = np.float32(transformed_image) if transformed_image.dtype != np.float32 else transformed_image.copy()
+        assert transformed_image.dtype != np.uint8
+        #if to_rgb:
+        cv2.cvtColor(transformed_image, cv2.COLOR_BGR2RGB, transformed_image)  # inplace
+
         results['img'] = transformed_image
         for key, mask in zip(keys, transformed_masks):
             results[key] = mask