Explicitly copying array in pil_to_tensor (#4566) (#4573)

* mExplicitly copying array in pil_to_tensor

* Update torchvision/transforms/functional.py

Co-authored-by: Nicolas Hug <[email protected]>

* Adding comments regarding implicit array deep copy in PILToTensor transform

* Update torchvision/transforms/transforms.py

Co-authored-by: Nicolas Hug <[email protected]>

Co-authored-by: Nicolas Hug <[email protected]>
Co-authored-by: Vasilis Vryniotis <[email protected]>

Co-authored-by: Joao Gomes <[email protected]>
Co-authored-by: Nicolas Hug <[email protected]>

Author: 3 people

torchvision/transforms/functional.py

@@ -158,6 +158,10 @@ def pil_to_tensor(pic):
 
     See :class:`~torchvision.transforms.PILToTensor` for more details.
 
+    .. note::
+
+        A deep copy of the underlying array is performed.
+
     Args:
         pic (PIL Image): Image to be converted to tensor.
 
@@ -174,7 +178,7 @@ def pil_to_tensor(pic):
         return torch.as_tensor(nppic)
 
     # handle PIL Image
-    img = torch.as_tensor(np.asarray(pic))
+    img = torch.as_tensor(np.array(pic, copy=True))
     img = img.view(pic.size[1], pic.size[0], len(pic.getbands()))
     # put it from HWC to CHW format
     img = img.permute((2, 0, 1))

torchvision/transforms/transforms.py

@@ -109,6 +109,10 @@ class PILToTensor:
 
     def __call__(self, pic):
         """
+        .. note::
+
+            A deep copy of the underlying array is performed.
+
         Args:
             pic (PIL Image): Image to be converted to tensor.