实现图像中斜放物体转正的操作

实现图像中斜放物体转正的操作

我用的是pycharm中的python3.x，因为自己研究过程中会存在物体斜放的情况，所以需要通过程序将物体进行摆正后处理。方法比较简单，利用的是物体旋转的外接矩形的性质。

以上图为例，图像中物体是向右斜放的，我们所需要做的是将物体“摆正”。

import cv2
import numpy as np

#读取原图像
img=cv2.imread(‘C:/Users/86185/Desktop/8.2.jpg’)
#读取灰度图像
img1 = cv2.imread(‘C:/Users/86185/Desktop/8.2.jpg’,0)
#使用中值滤波
img2 = cv2.medianBlur(img1,39)
#二值化
ret,thresh1=cv2.threshold(img2,50,255,cv2.THRESH_BINARY)
#形态学运算中的开运算（opening）:先腐蚀再膨胀
kernel = np.ones((5,5),np.uint8)
opening = cv2.morphologyEx(thresh1, cv2.MORPH_OPEN, kernel)
#根据阈值识别出分叉类型的胡萝卜
_, contours, hierarchy = cv2.findContours(opening, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)
cnt = contours[0]
rect = cv2.minAreaRect(cnt)
box = cv2.boxPoints(rect)
box = np.int0(box)
cv2.drawContours(img, [box], 0, (0, 0, 255), 2)
x,y,w,h = cv2.boundingRect(cnt) #（x,y）是旋转的边界矩形左上角的点，w ,h分别是宽和高

o1,o2 = rect[0]
o1 = int(o1)
if w < h:
angle = int(rect[2])
M = cv2.getRotationMatrix2D((o1, o2), angle, 1)
rows, cols = img.shape[:2]
print(‘left’)
# print(rows,cols)
dst = cv2.warpAffine(img, M, (cols, rows))
if w > h:
angle = 90 + int(rect[2])
M = cv2.getRotationMatrix2D((o1, o2), angle, 1)
rows, cols = img.shape[:2]
# print(rows,cols)
dst = cv2.warpAffine(img, M, (cols, rows))
print(‘right’)

根据宽和高的大小判断物体向哪个方向“歪”的。
运行程序得到的图像如下图：

另外一张向左歪的图像，试了后也可适用，效果还行。