Upload processor

image_processing_basnet.py

@@ -17,7 +17,7 @@ class RescaleT(object):
         assert isinstance(output_size, (int, tuple))
         self.output_size = output_size
 
-    def __call__(self, sample):
+    def __call__(self, sample) -> Dict[str, np.ndarray]:
         image, label = sample["image"], sample["label"]
 
         h, w = image.shape[:2]
@@ -61,7 +61,7 @@ class RescaleT(object):
 class ToTensorLab(object):
     """Convert ndarrays in sample to Tensors."""
 
-    def __init__(self, flag=0):
+    def __init__(self, flag: int = 0) -> None:
         self.flag = flag
 
     def __call__(self, sample):
@@ -74,7 +74,6 @@ class ToTensorLab(object):
         else:
             label = label / np.max(label)
 
-        # print('self.flag:', self.flag) # Default: 0
         # change the color space
         if self.flag == 2:  # with rgb and Lab colors
             tmpImg = np.zeros((image.shape[0], image.shape[1], 6))