未验证 提交 5546f6ac 编写于 作者: X Xuuu 提交者: GitHub

Fix bugs in the resize_short and simple_transform funciton

The resize_short function used the cv2.resize function in a wrong way.
The simple_transform function repeated the code uncorrectly.
上级 866fcb0c
...@@ -182,7 +182,7 @@ def resize_short(im, size): ...@@ -182,7 +182,7 @@ def resize_short(im, size):
h_new = size * h / w h_new = size * h / w
else: else:
w_new = size * w / h w_new = size * w / h
im = cv2.resize(im, (h_new, w_new), interpolation=cv2.INTER_CUBIC) im = cv2.resize(im, (w_new, h_new), interpolation=cv2.INTER_CUBIC)
return im return im
...@@ -324,7 +324,6 @@ def simple_transform(im, ...@@ -324,7 +324,6 @@ def simple_transform(im,
if np.random.randint(2) == 0: if np.random.randint(2) == 0:
im = left_right_flip(im, is_color) im = left_right_flip(im, is_color)
else: else:
im = center_crop(im, crop_size, is_color)
im = center_crop(im, crop_size, is_color=is_color) im = center_crop(im, crop_size, is_color=is_color)
if len(im.shape) == 3: if len(im.shape) == 3:
im = to_chw(im) im = to_chw(im)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册