Create image_processing.py

Hetul79 · web-flow · commit 3525d6b47bbf · 2020-10-10T22:44:59.000+05:30
diff --git a/image_processing.py b/image_processing.py
@@ -0,0 +1,15 @@
+import sys
+import os
+from PIL import Image
+
+image_folder = sys.argv[1]
+output_folder = sys.argv[2]
+
+if  not os.path.exists(output_folder):
+    os.makedirs(output_folder)
+
+for filename in os.listdir(image_folder):
+    img = Image.open(f'{image_folder}{filename}')  
+    clean_name = os.path.splitext(filename)[0]
+    img.save(f'{output_folder}{clean_name}.png','png')
+    print('all done')
