[imgaug] check xrange==yrange only when using ratio (fix #628)

tensorpack · Feb 5, 2018 · 0e5299b · 0e5299b · eyaler · Feb 5, 2018
1 parent fe04987
commit 0e5299b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tensorpack/dataflow/imgaug/misc.py b/tensorpack/dataflow/imgaug/misc.py
@@ -121,15 +121,17 @@ def __init__(self, xrange, yrange, minimum=(0, 0), aspect_ratio_thres=0.15,
         """
         super(RandomResize, self).__init__()
         assert aspect_ratio_thres >= 0
-        if aspect_ratio_thres == 0:
-            assert xrange == yrange
         self._init(locals())
 
         def is_float(tp):
             return isinstance(tp[0], float) or isinstance(tp[1], float)
+
         assert is_float(xrange) == is_float(yrange), "xrange and yrange has different type!"
         self._is_scale = is_float(xrange)
 
+        if self._is_scale and aspect_ratio_thres == 0:
+            assert xrange == yrange
+
     def _get_augment_params(self, img):
         cnt = 0
         h, w = img.shape[:2]