mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-02-21 11:18:01 +00:00
31 lines
652 B
Python
31 lines
652 B
Python
|
from PIL import Image
|
||
|
|
from tqdm import tqdm
|
||
|
|
import os
|
||
|
|
from multiprocessing import Pool
|
||
|
|
root_dir = "FFHQ/images1024x1024/"
|
||
|
|
files = os.listdir(root_dir)
|
||
|
|
|
||
|
|
def resize(file, size, folder_to_save):
|
||
|
|
image = Image.open(root_dir + file).resize((size, size), Image.LANCZOS)
|
||
|
|
image.save(folder_to_save+file, quality=100)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
for img_size in [4, 8, 512, 1024]:
|
||
|
|
folder_name = "FFHQ_"+str(img_size)+"/images/"
|
||
|
|
if not os.path.isdir(folder_name):
|
||
|
|
os.makedirs(folder_name)
|
||
|
|
|
||
|
|
data = [(file, img_size, folder_name) for file in files]
|
||
|
|
pool = Pool()
|
||
|
|
pool.starmap(resize, data)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|