Python批量处理图像【数据扩增...】

1 批量处理图像-翻转、旋转

Code

import cv2
import os
import numpy as np
from tqdm import tqdm

def read_path(file_path,save_path):
    # 创建文件夹
    if not os.path.exists(save_path):
        os.makedirs(save_path)
        
    #遍历该目录下的所有图片文件
    for filename in tqdm(os.listdir(file_path)):# 增加进度条
        img = os.path.join(file_path,filename)

        img=cv2.imread(img)
        img = np.rot90(img,-1)  # 将img顺时旋转90°，为正值时逆时针旋转
        img = np.flipud(img)          #将矩阵上下翻转，输入图像的话，翻转后生成新的图像

        output_path = os.path.join(save_path, filename)
        cv2.imwrite(output_path,img)

if __name__ == '__main__':
    img_path="C:/Users/XUB/Desktop/A"
    save_path="C:/Users/XUB/Desktop/TA"
    read_path(img_path,save_path)