Updated readme

README.md

@@ -2,4 +2,8 @@
 license: mit
 ---
 
-This serves as a repo for sharing some tools that I made for processing Lora training dataset.
+# **This serves as a repo for sharing some tools that I made for processing Lora training dataset.**
+
+## sort_image.py
+Discription: A simple python script that puts images with certain format into different folder for upscaling based on images' resolution. Only images in current directory would be sorted (images in sub-directories would not be sorted).
+Requirement: Python 3.*

sort_image.py

@@ -5,10 +5,7 @@ import shutil
 
 # Global Vars
 # Folders' name. Each folder serves as one kind of level for image upscale
-UPSCALE_FOLDERS = ['X2', 'X3', 'X4']
-X4_FOLDER = 'X4'
-X3_FOLDER = 'X3'
-X2_FOLDER = 'X2'
+UPSCALE_FOLDERS = {'X2_FOLDER':'X2', 'X3_FOLDER':'X3', 'X4_FOLDER':'X4'}
 # Image's required format 
 imgFormat = '.png'
 # Min and Max limit for different upscale level
@@ -35,17 +32,17 @@ def getAllImgs():
             #print('Width: ' + str(imgWid) + ' Height: ' + str(imgHgt))
             res = imgWid*imgHgt
             if(res <= X4_Max):
-                shutil.move(eachFile, X4_FOLDER)
+                shutil.move(eachFile, UPSCALE_FOLDERS['X4_FOLDER'])
             elif(res > X3_Min and res <= X3_Max):
-                shutil.move(eachFile, X3_FOLDER)
+                shutil.move(eachFile, UPSCALE_FOLDERS['X3_FOLDER'])
             elif(res > X2_Min and res <= X2_Max):
-                shutil.move(eachFile, X2_FOLDER)
+                shutil.move(eachFile, UPSCALE_FOLDERS['X2_FOLDER'])
     return imgs
 
 
 #############################################
 for folder in UPSCALE_FOLDERS:
-    if not os.path.exists(folder):
-        os.makedirs(folder)
+    if not os.path.exists(UPSCALE_FOLDERS[folder]):
+        os.makedirs(UPSCALE_FOLDERS[folder])
 
 print(getAllImgs())