Replaced example usage of ToTensor() by PILToTensor() + ConvertImageDtype() (#4494)

eb2fb253 · Prabhat Roy · GitHub · 7ea38b32 · eb2fb253
Unverified Commit eb2fb253 authored Sep 28, 2021 by Prabhat Roy Committed by GitHub Sep 28, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

torchvision/transforms/transforms.py torchvision/transforms/transforms.py +4 -2

No files found.
--- a/torchvision/transforms/transforms.py
+++ b/torchvision/transforms/transforms.py
@@ -35,7 +35,8 @@ class Compose:
    Example:
        >>> transforms.Compose([
        >>>     transforms.CenterCrop(10),
-        >>>     transforms.ToTensor(),
+        >>>     transforms.PILToTensor(),
+        >>>     transforms.ConvertImageDtype(torch.float),
        >>> ])

    .. note::
@@ -1570,7 +1571,8 @@ class RandomErasing(torch.nn.Module):
    Example:
        >>> transform = transforms.Compose([
        >>>   transforms.RandomHorizontalFlip(),
-        >>>   transforms.ToTensor(),
+        >>>   transforms.PILToTensor(),
+        >>>   transforms.ConvertImageDtype(torch.float),
        >>>   transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)),
        >>>   transforms.RandomErasing(),
        >>> ])