【OpenCV】54 二值图像分析—对轮廓圆与椭圆拟合

54 二值图像分析—对轮廓圆与椭圆拟合

代码

import cv2 as cv
import numpy as np

def canny_demo(image):
    t = 80
    canny_output = cv.Canny(image, t, t * 2)
    cv.imshow("canny_output", canny_output)
    return canny_output

src = cv.imread("../images/stuff.jpg")
cv.namedWindow("input", cv.WINDOW_AUTOSIZE)
cv.imshow("input", src)
binary = canny_demo(src)
k = np.ones((3, 3), dtype=np.uint8)
binary = cv.morphologyEx(binary, cv.MORPH_DILATE, k) #此处运行了形态学处理中的膨胀，在后面会给予介绍

# 轮廓发现
contours, hierarchy = cv.findContours(binary, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_SIMPLE)
for c in range(len(contours)):
    # 椭圆拟合
    (cx, cy), (a, b), angle = cv.fitEllipse(contours[c])
    # 绘制椭圆
    cv.ellipse(src, (np.int32(cx), np.int32(cy)),
               (np.int32(a/2), np.int32(b/2)), angle, 0, 360, (0, 0, 255), 2, 8, 0)

# 显示
cv.imshow("contours_analysis", src)
cv.waitKey(0)
cv.destroyAllWindows()

实验结果

解释

有时候我们需要对找到的轮廓点进行拟合，生成一个拟合的圆形或者椭圆，以便我们对轮廓进行更进一步的处理，满足我们对最终轮廓形状的判断，OpenCV对轮廓进行圆形或者椭圆拟合的API函数如下：

retval = cv.fitEllipse(points)

参数points是轮廓点，
输出RotatedRect包含下面三个信息

• 拟合之后圆或者椭圆的中心位置、
• 长轴与短轴的直径
• 角度
  然后我们就可以根据得到拟合信息绘制椭圆、当长轴与短轴相等的时候就是圆。

---

所有内容均来源于贾志刚老师的知识星球——OpenCV研习社，本文为个人整理学习，已获得贾老师授权，有兴趣、有能力的可以加入贾老师的知识星球进行深入学习。