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))