diff --git a/torchvision/prototype/datapoints/_datapoint.py b/torchvision/prototype/datapoints/_datapoint.py index 5c50542b07d..848808d0250 100644 --- a/torchvision/prototype/datapoints/_datapoint.py +++ b/torchvision/prototype/datapoints/_datapoint.py @@ -230,7 +230,7 @@ def elastic( ) -> Datapoint: return self - def to_grayscale(self, num_output_channels: int = 1) -> Datapoint: + def rgb_to_grayscale(self, num_output_channels: int = 1) -> Datapoint: return self def adjust_brightness(self, brightness_factor: float) -> Datapoint: diff --git a/torchvision/prototype/datapoints/_image.py b/torchvision/prototype/datapoints/_image.py index 0b2ab7453ff..56939bf14d9 100644 --- a/torchvision/prototype/datapoints/_image.py +++ b/torchvision/prototype/datapoints/_image.py @@ -169,7 +169,7 @@ def elastic( ) return Image.wrap_like(self, output) - def to_grayscale(self, num_output_channels: int = 1) -> Image: + def rgb_to_grayscale(self, num_output_channels: int = 1) -> Image: output = self._F.rgb_to_grayscale_image_tensor( self.as_subclass(torch.Tensor), num_output_channels=num_output_channels ) diff --git a/torchvision/prototype/datapoints/_video.py b/torchvision/prototype/datapoints/_video.py index 50f9110f40c..6c24197a9ca 100644 --- a/torchvision/prototype/datapoints/_video.py +++ b/torchvision/prototype/datapoints/_video.py @@ -173,7 +173,7 @@ def elastic( ) return Video.wrap_like(self, output) - def to_grayscale(self, num_output_channels: int = 1) -> Video: + def rgb_to_grayscale(self, num_output_channels: int = 1) -> Video: output = self._F.rgb_to_grayscale_image_tensor( self.as_subclass(torch.Tensor), num_output_channels=num_output_channels ) diff --git a/torchvision/prototype/transforms/functional/_color.py b/torchvision/prototype/transforms/functional/_color.py index 719bd801e74..e1c8bb87cdd 100644 --- a/torchvision/prototype/transforms/functional/_color.py +++ b/torchvision/prototype/transforms/functional/_color.py @@ -46,7 +46,7 @@ def rgb_to_grayscale( if torch.jit.is_scripting() or is_simple_tensor(inpt): return rgb_to_grayscale_image_tensor(inpt, num_output_channels=num_output_channels) elif isinstance(inpt, datapoints._datapoint.Datapoint): - return inpt.to_grayscale(num_output_channels=num_output_channels) + return inpt.rgb_to_grayscale(num_output_channels=num_output_channels) elif isinstance(inpt, PIL.Image.Image): return rgb_to_grayscale_image_pil(inpt, num_output_channels=num_output_channels) else: