diff --git a/test/test_transforms.py b/test/test_transforms.py index 54aae796301..7e8320d6d6c 100644 --- a/test/test_transforms.py +++ b/test/test_transforms.py @@ -141,8 +141,9 @@ def test_randomresized_params(self): img = to_pil_image(img) size = 100 epsilon = 0.05 + min_scale = 0.25 for _ in range(10): - scale_min = round(random.random(), 2) + scale_min = max(round(random.random(), 2), min_scale) scale_range = (scale_min, scale_min + round(random.random(), 2)) aspect_min = max(round(random.random(), 2), epsilon) aspect_ratio_range = (aspect_min, aspect_min + round(random.random(), 2)) diff --git a/torchvision/transforms/transforms.py b/torchvision/transforms/transforms.py index 964504eb9dc..203dae345cd 100644 --- a/torchvision/transforms/transforms.py +++ b/torchvision/transforms/transforms.py @@ -645,7 +645,7 @@ def get_params(img, scale, ratio): w = int(round(math.sqrt(target_area * aspect_ratio))) h = int(round(math.sqrt(target_area / aspect_ratio))) - if w <= img.size[0] and h <= img.size[1]: + if 0 < w <= img.size[0] and 0 < h <= img.size[1]: i = random.randint(0, img.size[1] - h) j = random.randint(0, img.size[0] - w) return i, j, h, w