Commit 3ed48310 authored by Soumith Chintala's avatar Soumith Chintala Committed by GitHub
Browse files

Merge pull request #21 from pytorch/float

Fixes for inconsistency between / in Python 2 and 3
parents 5c39840c 0786baec
...@@ -111,8 +111,8 @@ class CenterCrop(object): ...@@ -111,8 +111,8 @@ class CenterCrop(object):
def __call__(self, img): def __call__(self, img):
w, h = img.size w, h = img.size
th, tw = self.size th, tw = self.size
x1 = int(round((w - tw) / 2)) x1 = int(round((w - tw) / 2.))
y1 = int(round((h - th) / 2)) y1 = int(round((h - th) / 2.))
return img.crop((x1, y1, x1 + tw, y1 + th)) return img.crop((x1, y1, x1 + tw, y1 + th))
...@@ -187,7 +187,7 @@ class RandomSizedCrop(object): ...@@ -187,7 +187,7 @@ class RandomSizedCrop(object):
for attempt in range(10): for attempt in range(10):
area = img.size[0] * img.size[1] area = img.size[0] * img.size[1]
target_area = random.uniform(0.08, 1.0) * area 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))) w = int(round(math.sqrt(target_area * aspect_ratio)))
h = int(round(math.sqrt(target_area / aspect_ratio))) h = int(round(math.sqrt(target_area / aspect_ratio)))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment