-
Notifications
You must be signed in to change notification settings - Fork 0
/
preprocess_images.py
30 lines (24 loc) · 1.22 KB
/
preprocess_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
from PIL import Image
import argparse
import os
def resize_images(path_img, path_output, size_x=512, size_y=512):
for l in os.listdir(path_img):
print('Artwork image file: {}'.format(l))
if not (l.endswith('.jpg') or l.endswith('.png')):
continue
img = Image.open(os.path.join(path_img, l))
print(img.size)
img_resized = img.resize(
(size_x,size_y)
)
img_resized.save(os.path.join(path_output, l))
if __name__ == '__main__':
argument_parser = argparse.ArgumentParser(
'Artwork image preprocessing, i.e. resizing for model'
)
argument_parser.add_argument('-pi', '--path_img', type=str, default='data/raw_images', help='Path to folder containing raw image')
argument_parser.add_argument('-po', '--path_output', type=str, default='./data/proc', help='Path where resized artwork images are saved to')
argument_parser.add_argument('-sx', '--size_x', type=int, default=512, help='New x size (width) of img')
argument_parser.add_argument('-sy', '--size_y', type=int, default=512, help='New y size (height) of img')
args = argument_parser.parse_args()
resize_images(args.path_img, args.path_output, args.size_x, args.size_y)