multimodal_aug_transform.py

class Transform:
  @abstractmethod
  def apply_image(self, img: np.ndarray):
    
  @abstractmethod
  def apply_coords(self, coords: np.ndarray):
    
  def apply_segmentation(self, segmentation: np.ndarray) -> np.ndarray:
    return self.apply_image(segmentation)
    
  def apply_box(self, box: np.ndarray) -> np.ndarray:
    ...
    return trans_boxes
    
  def apply_polygons(self, polygons: list) -> list:
    return [self.apply_coords(p) for p in polygons]