| 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") |