在pycharm使用opencv完成矩形识别并返回坐标

import cv2
import numpy as np

#定义形状检测函数
def ShapeDetection(img):
    contours, _ = cv2.findContours(img,cv2.RETR_LIST,cv2.CHAIN_APPROX_NOME)  #寻找轮廓点
    for obj in contours:
        area = cv2.contourArea(obj)  #计算轮廓内区域的面积
        # cv2.drawContours(imgContour, obj, -1, (255, 0, 0), 4)  #绘制轮廓线
        perimeter = cv2.arcLength(obj,True)  #计算轮廓周长
        approx = cv2.approxPolyDP(obj,0.03*perimeter,True)  #获取轮廓角点坐标
        CornerNum = len(approx)   #轮廓角点的数量

        #轮廓对象分类

        if (CornerNum == 4) :
            x, y, w, h = cv2.boundingRect(approx)  # 获取坐标值和宽度、高度
            if w > 40 and h > 30:
                cv2.drawContours(imgContour, [approx], -1, (255, 0, 0), 4)  #绘制轮廓线
                for i in range(4):
                    # 计算角点的坐标
                    pt = tuple(approx[i][0])
                    cv2.circle(imgContour, pt, 5, (0, 255, 0), -1)  # 绘制角点

                    # 可选:显示角点的坐标
                    cv2.putText(imgContour, str(pt), (pt[0] + 10, pt[1] + 10),
                                cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 255), 1)
                    print(approx[i][0])

                    # 绘制矩形类型和名称
                cv2.putText(imgContour, "Rectangel", (x + (w // 2), y + (h // 2)),
                            cv2.FONT_HERSHEY_COMPLEX, 0.6, (0, 0, 0), 1)

path = 'B.png'
img = cv2.imread(path)
imgContour = img.copy()

imgGray = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)  #转灰度图
imgBlur = cv2.GaussianBlur(imgGray,(5,5),1)  #高斯模糊
imgCanny = cv2.Canny(imgBlur,60,60)  #Canny算子边缘检测
ShapeDetection(imgCanny)  #形状检测

# cv2.imshow("Original img", img)
# cv2.imshow("imgGray", imgGray)
# cv2.imshow("imgBlur", imgBlur)
# cv2.imshow("imgCanny", imgCanny)
cv2.imshow("shape Detection", imgContour)


cv2.waitKey(0)

结果如下
  • 16
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值