Python图像翻转[附批量操作函数]
new_image = cv.flip(original_image, 0)
new_image = original_image[::-1]
new_image = cv.flip(original_image, 1)
new_image = original_image[:,::-1]
new_image = cv.flip(cv.flip(original_image, 0),1)
new_image = original_image[::-1]
new_image = new_image[:,::-1]
分享一个自用函数,可进行图像的单张和批量翻转(包括以上三种操作)
import numpy as np
import cv2 as cv
import os
"""
<mode1>:
'single':对单个图像进行操作。'folder':对文件夹内所有图像批量操作。
<mode2>:
'UD':图像上下翻转。'LR':图像左右翻转。'T':图像转置。
------------------------------------------------------------------
<singlepath>:单个图像的绝对路径。
<original_path>:文件夹批量操作,文件夹的绝对路径。
<new_path>:文件夹批量操作,新生成文件存储文件夹的绝对路径。
"""
def image_flip(mode1, mode2, singlepath=None, original_path=None, new_path=None):
if new_path is not None and os.path.exists(new_path) is False:
os.makedirs(new_path)
if mode1 == 'folder':
for name in os.listdir(original_path):
original_image = cv.imread(original_path + '/' + name)
savepath = (new_path+'/' + name[:name.index('.')] + '_' + str(mode2) + name[name.index('.'):])
if mode2 == 'UD':
new_image = cv.flip(original_image, 0)
cv.imwrite(savepath, new_image)
elif mode2 == 'LR':
new_image = cv.flip(original_image, 1)
cv.imwrite(savepath, new_image)
elif mode2 == 'T':
new_image = cv.flip(original_image, -1)
cv.imwrite(savepath, new_image)
else:
print("mode2可选参数为:\n'UD':上下翻转\n'LR':左右翻转\n'T' :转置\n请输入正确参数!!!")
break
elif mode1 == 'single':
original_image = cv.imread(singlepath)
savepath = singlepath[:singlepath.index('.')] + str(mode2) + singlepath[singlepath.index('.'):]
if mode2 == 'UD':
new_image = cv.flip(original_image, 0)
cv.imwrite(savepath, new_image)
if mode2 == 'LR':
new_image = cv.flip(original_image, 1)
cv.imwrite(savepath, new_image)
if mode2 == 'T':
new_image = cv.flip(original_image, -1)
cv.imwrite(savepath, new_image)
else:
print("mode1可选参数为:\n'folder':批量处理\n'single':处理单张\n请输入正确参数!!!")
os.rmdir(new_path)
image_flip('single', 'T', r'C:\..\..\..\..\banana_sample.jpg')
image_flip('single', 'LR', r'C:\..\..\..\..\banana_sample.jpg')
image_flip('single', 'T', r'C:\..\..\..\..\banana_sample.jpg')
image_flip('folder', 'UD', original_path='E:/../../1', new_path='E:/../../2')
Result:
单图:
批量: