| 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) |
