|
|
@ -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,
|
|
|
|
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)
|
|
|
|