(fix) fix RandomCrop get_params output (#289)

12a37ba7 · Yu Luo · Alykhan Tejani · 9fc7edfb · 12a37ba7
Commit 12a37ba7 authored Oct 12, 2017 by Yu Luo Committed by Alykhan Tejani Oct 12, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/transforms.py torchvision/transforms.py +1 -1

No files found.
--- a/torchvision/transforms.py
+++ b/torchvision/transforms.py
@@ -749,7 +749,7 @@ class RandomCrop(object):
        w, h = img.size
        th, tw = output_size
        if w == tw and h == th:
-            return img
+            return 0, 0, h, w
        i = random.randint(0, h - th)
        j = random.randint(0, w - tw)