Merge pull request pytorch#21 from pytorch/float

Fixes for inconsistency between / in Python 2 and 3
graingert · Jan 3, 2017 · 3ed4831 · 3ed4831
2 parents 5c39840 + 0786bae
commit 3ed4831
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/torchvision/transforms.py b/torchvision/transforms.py
@@ -41,7 +41,7 @@ def __call__(self, pic):
         return img.float().div(255)
 
 class ToPILImage(object):
-    """ Converts a torch.*Tensor of range [0, 1] and shape C x H x W 
+    """ Converts a torch.*Tensor of range [0, 1] and shape C x H x W
     or numpy ndarray of dtype=uint8, range[0, 255] and shape H x W x C
     to a PIL.Image of range [0, 255]
     """
@@ -111,8 +111,8 @@ def __init__(self, size):
     def __call__(self, img):
         w, h = img.size
         th, tw = self.size
-        x1 = int(round((w - tw) / 2))
-        y1 = int(round((h - th) / 2))
+        x1 = int(round((w - tw) / 2.))
+        y1 = int(round((h - th) / 2.))
         return img.crop((x1, y1, x1 + tw, y1 + th))
 
 
@@ -187,7 +187,7 @@ def __call__(self, img):
         for attempt in range(10):
             area = img.size[0] * img.size[1]
             target_area = random.uniform(0.08, 1.0) * area
-            aspect_ratio = random.uniform(3 / 4, 4 / 3)
+            aspect_ratio = random.uniform(3. / 4, 4. / 3)
 
             w = int(round(math.sqrt(target_area * aspect_ratio)))
             h = int(round(math.sqrt(target_area / aspect_ratio)))