Python实现批量图像镜像翻转

'''
Python批量实现图像镜像翻转
函数：DataAugment（）
函数功能：扩大数据量
输入参数：dir_path----图片库路径
'''


import cv2
import os
import copy

def DataAugment(dir_path):
    if not os.path.exists(dir_path):
        print('路径不存在')
    else:
        dirs = os.listdir(dir_path)
    
        for subdir in dirs:
            sub_dir = dir_path + '/' + subdir
            img = cv2.imread(sub_dir)
          
            
            size = img.shape#获得图像的形状 
            
            
            iLR = copy.deepcopy(img)#获得一个和原始图像相同的图像，注意这里要使用深度复制
            h = size[0]
            w = size[1]
          
            for i in range(h):#元素循环
                for j in range(w):
                    iLR[i,w-1-j]=img[i,j]#注意这里的公式没，是不是恍然大悟了（修改这里）
                    
            new_name ="%s"%(sub_dir)
            cv2.imwrite(new_name,iLR)

            
            
    print('done')
                
                
                
                
                
DataAugment(r"C:\Users\Sen\Desktop\aaa" )
#地址中不能有中文字符，否则会报错