From 5a40f537cce1f74eedab36680b5bad6ab6478093 Mon Sep 17 00:00:00 2001 From: Colombo Date: Sat, 4 Jul 2020 07:58:53 +0400 Subject: [PATCH] . --- core/imagelib/warp.py | 6 +++--- core/leras/ops/__init__.py | 3 +-- 2 files changed, 4 insertions(+), 5 deletions(-) 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