'''
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" )
#地址中不能有中文字符,否则会报错
Python实现批量图像镜像翻转
最新推荐文章于 2024-05-17 05:55:51 发布