[FBcode->GH] Fix to_tensor for accimage backend (#3439)

67681a76 · Francisco Massa · GitHub · a7b4bfd6 · 67681a76
Unverified Commit 67681a76 authored Feb 23, 2021 by Francisco Massa Committed by GitHub Feb 23, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

No files found.
--- a/torchvision/transforms/functional.py
+++ b/torchvision/transforms/functional.py
@@ -119,9 +119,9 @@ def to_tensor(pic):
            return img

    if accimage is not None and isinstance(pic, accimage.Image):
-        nppic = np.zeros([pic.channels, pic.height, pic.width], dtype=default_float_dtype)
+        nppic = np.zeros([pic.channels, pic.height, pic.width], dtype=np.float32)
        pic.copyto(nppic)
-        return torch.from_numpy(nppic)
+        return torch.from_numpy(nppic).to(dtype=default_float_dtype)

    # handle PIL Image
    if pic.mode == 'I':