直接上代码
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")
生成的目录如下所示