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

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

我用的是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’)

根据宽和高的大小判断物体向哪个方向“歪”的。
运行程序得到的图像如下图:
在这里插入图片描述
另外一张向左歪的图像,试了后也可适用,效果还行。
在这里插入图片描述
在这里插入图片描述

  • 0
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值