jim48949
commited on
Commit
•
285a52d
1
Parent(s):
9241a1d
Updated readme
Browse files- README.md +5 -1
- sort_image.py +6 -9
README.md
CHANGED
@@ -2,4 +2,8 @@
|
|
2 |
license: mit
|
3 |
---
|
4 |
|
5 |
-
This serves as a repo for sharing some tools that I made for processing Lora training dataset
|
|
|
|
|
|
|
|
|
|
2 |
license: mit
|
3 |
---
|
4 |
|
5 |
+
# **This serves as a repo for sharing some tools that I made for processing Lora training dataset.**
|
6 |
+
|
7 |
+
## sort_image.py
|
8 |
+
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).
|
9 |
+
Requirement: Python 3.*
|
sort_image.py
CHANGED
@@ -5,10 +5,7 @@ import shutil
|
|
5 |
|
6 |
# Global Vars
|
7 |
# Folders' name. Each folder serves as one kind of level for image upscale
|
8 |
-
UPSCALE_FOLDERS =
|
9 |
-
X4_FOLDER = 'X4'
|
10 |
-
X3_FOLDER = 'X3'
|
11 |
-
X2_FOLDER = 'X2'
|
12 |
# Image's required format
|
13 |
imgFormat = '.png'
|
14 |
# Min and Max limit for different upscale level
|
@@ -35,17 +32,17 @@ def getAllImgs():
|
|
35 |
#print('Width: ' + str(imgWid) + ' Height: ' + str(imgHgt))
|
36 |
res = imgWid*imgHgt
|
37 |
if(res <= X4_Max):
|
38 |
-
shutil.move(eachFile, X4_FOLDER)
|
39 |
elif(res > X3_Min and res <= X3_Max):
|
40 |
-
shutil.move(eachFile, X3_FOLDER)
|
41 |
elif(res > X2_Min and res <= X2_Max):
|
42 |
-
shutil.move(eachFile, X2_FOLDER)
|
43 |
return imgs
|
44 |
|
45 |
|
46 |
#############################################
|
47 |
for folder in UPSCALE_FOLDERS:
|
48 |
-
if not os.path.exists(folder):
|
49 |
-
os.makedirs(folder)
|
50 |
|
51 |
print(getAllImgs())
|
|
|
5 |
|
6 |
# Global Vars
|
7 |
# Folders' name. Each folder serves as one kind of level for image upscale
|
8 |
+
UPSCALE_FOLDERS = {'X2_FOLDER':'X2', 'X3_FOLDER':'X3', 'X4_FOLDER':'X4'}
|
|
|
|
|
|
|
9 |
# Image's required format
|
10 |
imgFormat = '.png'
|
11 |
# Min and Max limit for different upscale level
|
|
|
32 |
#print('Width: ' + str(imgWid) + ' Height: ' + str(imgHgt))
|
33 |
res = imgWid*imgHgt
|
34 |
if(res <= X4_Max):
|
35 |
+
shutil.move(eachFile, UPSCALE_FOLDERS['X4_FOLDER'])
|
36 |
elif(res > X3_Min and res <= X3_Max):
|
37 |
+
shutil.move(eachFile, UPSCALE_FOLDERS['X3_FOLDER'])
|
38 |
elif(res > X2_Min and res <= X2_Max):
|
39 |
+
shutil.move(eachFile, UPSCALE_FOLDERS['X2_FOLDER'])
|
40 |
return imgs
|
41 |
|
42 |
|
43 |
#############################################
|
44 |
for folder in UPSCALE_FOLDERS:
|
45 |
+
if not os.path.exists(UPSCALE_FOLDERS[folder]):
|
46 |
+
os.makedirs(UPSCALE_FOLDERS[folder])
|
47 |
|
48 |
print(getAllImgs())
|