forked from qqwweee/keras-yolo3
-
Notifications
You must be signed in to change notification settings - Fork 9
/
resize_images.py
43 lines (30 loc) · 1.14 KB
/
resize_images.py
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
import os
import sys
from glob import glob
from PIL import Image
def resize_images(images_dir, image_save_dir, image_size):
os.makedirs(image_save_dir, exist_ok=True)
img_paths = glob(os.path.join(images_dir, '*'))
for img_path in img_paths:
# resize
image = Image.open(img_path)
rgb_im = image.convert('RGB')
rgb_im.thumbnail([image_size,image_size])
# make background
back_ground = Image.new("RGB", (image_size,image_size), color=(255,255,255))
back_ground.paste(rgb_im)
# make path
save_path = os.path.join(image_save_dir, os.path.basename(img_path))
end_index = save_path.rfind('.')
save_path = save_path[0:end_index]+'.jpg'
print('save',save_path)
back_ground.save(save_path,quality=95,format='JPEG')
def _main():
images_dir = 'images/' # input directory
image_save_dir = 'resize_image/' # output directory
image_size = 320
if len(sys.argv) > 1:
image_size = int(sys.argv[1])
resize_images(images_dir=images_dir, image_save_dir=image_save_dir, image_size=image_size)
if __name__ == '__main__':
_main()