Update functional.py (#421)

if on dtype -> elif on dtype

Update functional.py (#421)
if on dtype -> elif on dtype
13d1a6cc · vfdev · Soumith Chintala · f1b79d90 · 13d1a6cc
Commit 13d1a6cc authored Feb 15, 2018 by vfdev Committed by Soumith Chintala Feb 14, 2018
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
@@ -115,9 +115,9 @@ def to_pil_image(pic, mode=None):
        npimg = npimg[:, :, 0]
        if npimg.dtype == np.uint8:
            expected_mode = 'L'
-        if npimg.dtype == np.int16:
+        elif npimg.dtype == np.int16:
            expected_mode = 'I;16'
-        if npimg.dtype == np.int32:
+        elif npimg.dtype == np.int32:
            expected_mode = 'I'
        elif npimg.dtype == np.float32:
            expected_mode = 'F'