在做图像分割的问题的时候,用了数据增强,代码分享如下:
场景:一次性对两条路径的图片做相同的数据增强处理
path1——原图路径,path2——掩膜图路径
new_path1——增强后原图路径,new_path2——增强后掩膜图路径
count是计数器,会自动保存在新增图片的名字里。
#翻转和旋转图像
def enhanceimage1(path1, path2, new_path1, new_path2):
count = 0 #计数器
for path in [path1, path2]:
count = 0 #计数器
pic = [[0]] * 4
for pic_path in os.listdir(path):
pic[0] = cv2.imread(path+'/'+pic_path) #读入图片
pic[1] = cv2.flip(pic[0], 1)#水平翻转
pic[2] = cv2.flip(pic[0], 0)#垂直翻转
pic[3] = cv2.flip(pic[0], -1)#水平垂直翻转
for i in range(4):
if path == path1:
cv2.imwrite(new_path1+'{}'.format(i+count) +