File size: 534 Bytes
b81c7ad |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import os
import PIL.Image
def rename_imgs(org_folder, renamed_folder, name):
i = 1
for file in os.listdir(org_folder):
img = PIL.Image.open(os.path.join(org_folder, file))
# img = img.resize((300,300))
# if img.mode == "RGBA":
# img = img.convert("RGB")
img.save(os.path.join(renamed_folder, name+str(i)+".jpg"))
i += 1
if __name__ == "__main__":
rename_imgs("real_small", "Real", "real_train")
rename_imgs("ai_small", "AI generated", "ai_train") |