就是这么简单,只需要安装几个依赖库即可
import os
import numpy as np
from PIL import Image
from tqdm import tqdm
Origin_JPEGImages_path = "/home/pw/Dataset/val/image"
Out_JPEGImages_path = "/home/pw/Dataset/val/image2image"
if __name__ == "__main__":
if not os.path.exists(Out_JPEGImages_path):
os.makedirs(Out_JPEGImages_path)
image_names = os.listdir(Origin_JPEGImages_path)
print("Traversing all images!!!")
for image_name in tqdm(image_names):
image = Image.open(os.path.join(Origin_JPEGImages_path, image_name))
image = image.convert('RGB')
image.save(os.path.join(Out_JPEGImages_path, os.path.splitext(image_name)[0] + '.jpg'))