python批量处理并保存目录下的图像文件

直接上代码

def save_img(path):  #传入存储的list
    for file in os.listdir(path):
        file_path = os.path.join(path, file)
        new_filePath = "D:/DeskTop/PCBData/data/PCB_DATASET/PCB_DATASET/images"+"/"+ path.split("/")[-1]+"_color_augmentations"

        if os.path.exists(new_filePath)==0:
            os.makedirs(new_filePath)

        if os.path.isdir(file_path):
            save_img(file_path)
        else:
            image = cv2.imread(file_path, 1)  # BGR
            image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
            aug = augment_flips_color(p=1)
            img_augment_flips_color = aug(image=image)['image']
            new_fileName = file.split(".jpg")[0] + "_color_augmentations.jpg"
            cv2.imwrite(os.path.join(new_filePath,new_fileName), img_augment_flips_color)  # 保存图片

save_img("D:/DeskTop/PCBData/data/PCB_DATASET/PCB_DATASET/images/Spur")

生成的目录如下所示