Python opencv图像镜像翻转

最近在训练CNN网络，需要对数据进行扩充，于是想做一下图像的镜像翻转，主要参考了博客：

http://www.cnblogs.com/xianglan/archive/2010/12/25/1916982.html 感谢博主！

     不过我发现博主使用的是opencv老版本了，现在大家都是cv2了，因此我对博主的代码进行了必要的修改。

      首先，说明一下，图像的镜像变化不改变图像的形状。图像的镜像变换分为：水平镜像，垂直镜像，对角镜像

       下面，简单说一下三种变换的简单公式：假设原图像的大小为H*W，i,j为原始图像第（i，j）元素，I,J为变换后（I,J）元素，则：

        水平镜像可按公式：

I = i

J = W - 1 - j

垂直镜像可按公式：

I = H - 1 - i

J = j

对角镜像可按公式：

I = H - 1 - i

J = W -1 - i

就是这么简答，贴下我的代码吧！

'''
函数：DataAugment（）
函数功能：扩大数据量
输入参数：dir_path----图片库路径
'''
def DataAugment(dir_path):
    if not os.path.exists(dir_path):
        print u'路径不存在'
    else:
        dirs = os.listdir(dir_path)
        for subdir in dirs:
            sub_dir = dir_path + '/' + subdir
            files = os.listdir(sub_dir)
            fileNum = len(files)
            if fileNum > 25:
                continue
            num=0
            for fr in files:
                suff = fr.split('.')[1]
                filename = sub_dir + '/' + fr
                img = cv2.imread(filename)
                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/%09d.%s"%(sub_dir,num,suff)
                num+=1
                cv2.imwrite(new_name,iLR)
                #cv2.imshow('image',iLR)
                #cv2.waitKey(0)

说明：我这里只实现了水平镜像翻转，要实现其他翻转，可以使用其他公式。