File size: 631 Bytes
8aa7dae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from PIL import Image
import os
from tqdm import tqdm
in_file = "E:\\data\\pokemon_classify\\images\\test\\"
out_file = "E:\\data\\pokemon_classify\\resize_image\\test\\"
dict_file = os.listdir(in_file)
def main(dict_file, in_file, out_file):
for file in tqdm(dict_file):
img = Image.open(in_file + file)
w,h = img.size
ratio = w/h
new_w = 640
new_h = int(new_w/ratio)
new_img = img.resize((new_w,new_h), Image.ANTIALIAS)
file = file[:-4] + '.jpg'
new_img.save(out_file + file, quality = 100)
if __name__ == '__main__':
main(dict_file, in_file, out_file) |