From 781b25d7da3db4ad6f86cfadf53da1e515c58777 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Fri, 28 Jul 2023 10:23:02 +0100 Subject: [PATCH 1/2] Update deprecation warning --- torchvision/transforms/v2/_deprecated.py | 2 +- torchvision/transforms/v2/functional/_deprecated.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvision/transforms/v2/_deprecated.py b/torchvision/transforms/v2/_deprecated.py index e900e853d2b..cbc5bd5b915 100644 --- a/torchvision/transforms/v2/_deprecated.py +++ b/torchvision/transforms/v2/_deprecated.py @@ -40,7 +40,7 @@ class ToTensor(Transform): def __init__(self) -> None: warnings.warn( "The transform `ToTensor()` is deprecated and will be removed in a future release. " - "Instead, please use `transforms.Compose([transforms.ToImageTensor(), transforms.ConvertImageDtype()])`." + "Instead, please use `v2.Compose([transforms.ToImageTensor(), v2.ToDtype(torch.float32, scale=True)])`." ) super().__init__() diff --git a/torchvision/transforms/v2/functional/_deprecated.py b/torchvision/transforms/v2/functional/_deprecated.py index c9a0f647e60..99097aecc66 100644 --- a/torchvision/transforms/v2/functional/_deprecated.py +++ b/torchvision/transforms/v2/functional/_deprecated.py @@ -11,7 +11,7 @@ def to_tensor(inpt: Any) -> torch.Tensor: warnings.warn( "The function `to_tensor(...)` is deprecated and will be removed in a future release. " - "Instead, please use `to_image_tensor(...)` followed by `convert_image_dtype(...)`." + "Instead, please use `to_image_tensor(...)` followed by `to_dtype(..., dtype=torch.float32, scale=True)`." ) return _F.to_tensor(inpt) From 16626792f31ad5e3b29427abff0450f982de3c10 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Fri, 28 Jul 2023 10:31:14 +0100 Subject: [PATCH 2/2] update docstring --- torchvision/transforms/v2/_deprecated.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/transforms/v2/_deprecated.py b/torchvision/transforms/v2/_deprecated.py index cbc5bd5b915..1cb135a3062 100644 --- a/torchvision/transforms/v2/_deprecated.py +++ b/torchvision/transforms/v2/_deprecated.py @@ -16,7 +16,7 @@ class ToTensor(Transform): .. warning:: :class:`v2.ToTensor` is deprecated and will be removed in a future release. - Please use instead ``transforms.Compose([transforms.ToImageTensor(), transforms.ConvertImageDtype()])``. + Please use instead ``v2.Compose([transforms.ToImageTensor(), v2.ToDtype(torch.float32, scale=True)])``. This transform does not support torchscript.