Create resize_image.py

resize_image.py

@@ -0,0 +1,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)