Browse Source

fix error handling for bigger crop size case

tags/v0.5.0-beta
Tinazhang tinazhang 5 years ago
parent
commit
33fe66fd1e
2 changed files with 4 additions and 0 deletions
  1. +3
    -0
      mindspore/dataset/transforms/vision/py_transforms_util.py
  2. +1
    -0
      tests/ut/python/dataset/test_random_crop.py

+ 3
- 0
mindspore/dataset/transforms/vision/py_transforms_util.py View File

@@ -455,6 +455,9 @@ def random_crop(img, size, padding, pad_if_needed, fill_value, padding_mode):
def _input_to_factor(img, size): def _input_to_factor(img, size):
img_width, img_height = img.size img_width, img_height = img.size
height, width = size height, width = size
if height > img_height or width > img_width:
raise ValueError("Crop size {} is larger than input image size {}".format(size, (img_height, img_width)))

if width == img_width and height == img_height: if width == img_width and height == img_height:
return 0, 0, img_height, img_width return 0, 0, img_height, img_width




+ 1
- 0
tests/ut/python/dataset/test_random_crop.py View File

@@ -280,6 +280,7 @@ def test_random_crop_04_py():
data.create_dict_iterator().get_next() data.create_dict_iterator().get_next()
except RuntimeError as e: except RuntimeError as e:
logger.info("Got an exception in DE: {}".format(str(e))) logger.info("Got an exception in DE: {}".format(str(e)))
assert "Crop size" in str(e)


def test_random_crop_05_c(): def test_random_crop_05_c():
""" """


Loading…
Cancel
Save