diff --git a/docs/source/transforms.rst b/docs/source/transforms.rst index 8e3c60085de..0e9b053fb72 100644 --- a/docs/source/transforms.rst +++ b/docs/source/transforms.rst @@ -209,6 +209,7 @@ Conversion v2.ToTensor PILToTensor v2.PILToTensor + v2.ToImageTensor ConvertImageDtype v2.ConvertImageDtype v2.ConvertDtype diff --git a/torchvision/transforms/v2/_type_conversion.py b/torchvision/transforms/v2/_type_conversion.py index b0743feb10d..504c5cc3d70 100644 --- a/torchvision/transforms/v2/_type_conversion.py +++ b/torchvision/transforms/v2/_type_conversion.py @@ -27,6 +27,13 @@ def _transform(self, inpt: PIL.Image.Image, params: Dict[str, Any]) -> torch.Ten class ToImageTensor(Transform): + """[BETA] Convert a tensor or an ndarray or PIL Image to :class:`~torchvision.datapoints.Image`. + + .. betastatus:: ToImageTensor transform + + This transform does not support torchscript. + """ + _transformed_types = (is_simple_tensor, PIL.Image.Image, np.ndarray) def _transform(