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)