python opencv图片旋转180度_Python opencv图像镜像翻转
最近在訓練CNN網絡,需要對數據進行擴充,于是想做一下圖像的鏡像翻轉,主要參考了博客:
不過我發現博主使用的是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)
說明:我這里只實現了水平鏡像翻轉,要實現其他翻轉,可以使用其他公式。
《新程序員》:云原生和全面數字化實踐50位技術專家共同創作,文字、視頻、音頻交互閱讀總結
以上是生活随笔為你收集整理的python opencv图片旋转180度_Python opencv图像镜像翻转的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: python请求post无返回结果_Py
- 下一篇: 游戏开发需要具备哪些技术_生鲜小程序需要