added copy=False to np.array constructor

6d2423c7 · Bodo Kaiser · Soumith Chintala · 991bad2f · 6d2423c7
Commit 6d2423c7 authored Mar 23, 2017 by Bodo Kaiser Committed by Soumith Chintala Mar 23, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

No files found.
--- a/torchvision/transforms.py
+++ b/torchvision/transforms.py
@@ -43,9 +43,9 @@ class ToTensor(object):
            return img.float().div(255)
        # handle PIL Image
        if pic.mode == 'I':
-            img = torch.from_numpy(np.array(pic, np.int32))
+            img = torch.from_numpy(np.array(pic, np.int32, copy=False))
        elif pic.mode == 'I;16':
-            img = torch.from_numpy(np.array(pic, np.int16))
+            img = torch.from_numpy(np.array(pic, np.int16, copy=False))
        else:
            img = torch.ByteTensor(torch.ByteStorage.from_buffer(pic.tobytes()))
        # PIL image mode: 1, L, P, I, F, RGB, YCbCr, RGBA, CMYK