File size: 1,564 Bytes
75770b3 285a52d 75770b3 d981987 75770b3 285a52d 75770b3 285a52d 75770b3 285a52d 75770b3 285a52d 75770b3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import PIL
from PIL import Image
import os
import shutil
# Global Vars
# Folders' name. Each folder serves as one kind of level for image upscale
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
X2_Min = 2000**2
X2_Max = 2600**2
X3_Min = 1500**2
X3_Max = X2_Min
X4_Max = X3_Min
############################################################################
def getImgRes(imgName):
curImg = PIL.Image.open(imgName)
curImgWid, curImgHgt = curImg.size
return curImgWid, curImgHgt
def getAllImgs():
files = [f for f in os.listdir() if os.path.isfile(f)]
imgs = []
for eachFile in files:
if len(eachFile) >= 4 and eachFile[-4:] == imgFormat:
#print("Added: ", eachFile)
imgs.append(eachFile)
imgWid, imgHgt = getImgRes(eachFile)
#print('Width: ' + str(imgWid) + ' Height: ' + str(imgHgt))
res = imgWid*imgHgt
if(res <= X4_Max):
shutil.move(eachFile, UPSCALE_FOLDERS['X4_FOLDER'])
elif(res > X3_Min and res <= X3_Max):
shutil.move(eachFile, UPSCALE_FOLDERS['X3_FOLDER'])
elif(res > X2_Min and res <= X2_Max):
shutil.move(eachFile, UPSCALE_FOLDERS['X2_FOLDER'])
return imgs
#############################################
for folder in UPSCALE_FOLDERS:
if not os.path.exists(UPSCALE_FOLDERS[folder]):
os.makedirs(UPSCALE_FOLDERS[folder])
print(getAllImgs()) |