diff --git a/core/imagelib/warp.py b/core/imagelib/warp.py index 6860413..9252532 100644 --- a/core/imagelib/warp.py +++ b/core/imagelib/warp.py @@ -7,9 +7,9 @@ def gen_warp_params (w, flip, rotation_range=[-10,10], scale_range=[-0.5, 0.5], rnd_state = np.random rw = None - if w < 16: + if w < 64: rw = w - w = 16 + w = 64 rotation = rnd_state.uniform( rotation_range[0], rotation_range[1] ) scale = rnd_state.uniform(1 +scale_range[0], 1 +scale_range[1]) @@ -51,7 +51,7 @@ def warp_by_params (params, img, can_warp, can_transform, can_flip, border_repli rw = params['rw'] if (can_warp or can_transform) and rw is not None: - img = cv2.resize(img, (16,16), interpolation=cv2_inter) + img = cv2.resize(img, (64,64), interpolation=cv2_inter) if can_warp: img = cv2.remap(img, params['mapx'], params['mapy'], cv2_inter ) diff --git a/core/leras/ops/__init__.py b/core/leras/ops/__init__.py index ab42fa1..ddf1c06 100644 --- a/core/leras/ops/__init__.py +++ b/core/leras/ops/__init__.py @@ -141,7 +141,7 @@ def resize2d_nearest(x, size=2): if size in [-1,0,1]: return x - """ + if size > 0: raise Exception("") else: @@ -150,7 +150,6 @@ def resize2d_nearest(x, size=2): else: x = x[:,::-size,::-size,:] return x - """ h = x.shape[nn.conv2d_spatial_axes[0]].value w = x.shape[nn.conv2d_spatial_axes[1]].value