def __call__(self, img, target): """ Args: img (PIL Image): Image to be converted to grayscale. Returns: PIL Image: Randomly grayscaled image. """ return F.to_grayscale(img, num_output_channels=self.num_output_channels), F.to_grayscale(target, num_output_channels=self.num_output_channels)
def __call__(self, img): """ Args: img (numpy ndarray): Image to be converted to grayscale. Returns: numpy ndarray: Randomly grayscaled image. """ return F.to_grayscale(img, num_output_channels=self.num_output_channels)
def __call__(self, img): """ Args: img (PIL Image): Image to be converted to grayscale. Returns: PIL Image: Randomly grayscaled image. """ num_output_channels = 1 if img.mode == 'L' else 3 if random.random() < self.p: return F.to_grayscale(img, num_output_channels=num_output_channels) return img