1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
| import cv2
import numpy as np
import os
tf_w = 640
tf_h = 480
def img_resize(img_name):
img = cv2.imdecode(np.fromfile(img_name, dtype=np.uint8), -1)
w = img.shape[1]
h = img.shape[0]
ratio = h / w
target_ratio = tf_h / tf_w
if max(img.shape[:2]) > tf_w:
interp = cv2.INTER_AREA
else:
interp = cv2.INTER_LINEAR
img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
if img.shape[:2] == (tf_w, tf_h):
img_rgb_border = img
elif ratio == target_ratio:
img_rgb_border = cv2.resize(img_rgb, (tf_w, int(tf_w * ratio)), interpolation=interp)
elif (w > h) & (ratio > target_ratio):
img_rgb = cv2.resize(img_rgb, (int(tf_h * ratio), tf_h), interpolation=interp)
img_rgb_border = cv2.copyMakeBorder(img_rgb, top=0, bottom=0, left=0, right=tf_w - img_rgb.shape[1],
borderType=cv2.BORDER_CONSTANT,
value=[128, 128, 128])
elif w > h:
img_rgb = cv2.resize(img_rgb, (tf_w, int(tf_w * ratio)), interpolation=interp)
img_rgb_border = cv2.copyMakeBorder(img_rgb, top=0, bottom=tf_h - img_rgb.shape[0], left=0, right=0,
borderType=cv2.BORDER_CONSTANT,
value=[128, 128, 128])
elif (w <= h) & (ratio > target_ratio) & (ratio <= 1):
img_rgb = cv2.resize(img_rgb, (int(tf_h * ratio), tf_h), interpolation=interp)
img_rgb_border = cv2.copyMakeBorder(img_rgb, top=0, bottom=0, left=0, right=tf_w - img_rgb.shape[1],
borderType=cv2.BORDER_CONSTANT,
value=[128, 128, 128])
elif int(tf_h * ratio) < tf_w:
img_rgb = cv2.rotate(img_rgb, cv2.ROTATE_90_CLOCKWISE)
img_rgb = cv2.resize(img_rgb, (int(tf_h * ratio), tf_h), interpolation=interp)
img_rgb_border = cv2.copyMakeBorder(img_rgb, top=0, bottom=0, left=0,
right=tf_w - img_rgb.shape[1],
borderType=cv2.BORDER_CONSTANT,
value=[128, 128, 128])
else:
img_rgb = cv2.rotate(img_rgb, cv2.ROTATE_90_CLOCKWISE)
img_rgb = cv2.resize(img_rgb, (tf_w, int(tf_w / ratio)), interpolation=interp)
img_rgb_border = cv2.copyMakeBorder(img_rgb, top=0, bottom=tf_h - img_rgb.shape[0], left=0, right=0,
borderType=cv2.BORDER_CONSTANT,
value=[128, 128, 128])
os.makedirs('./border', exist_ok=True)
cv2.imencode('.jpg',
cv2.cvtColor(img_rgb_border,
cv2.COLOR_RGB2BGR))[1].tofile('./border/%s-border.jpg' % os.path.basename(img_name)[:-4])
if __name__ == '__main__':
filepath = input('filepath=?\n')
img_resize(filepath)
|
