cv2.approxPolyDP函数实现轮廓线的多边形逼近-Toy模板网

这篇具有很好参考价值的文章主要介绍了cv2.approxPolyDP函数实现轮廓线的多边形逼近。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

实际项目需要拟合轮廓线，提取更贴合目标的四个点，于是找到了cv2.approxPolyDP函数。cv2.approxPolyDP() 使用了Douglas-Peucker算法，算法原理如下：

输入一组曲线点集合S，输出折线点集合T流程：

step1、设阈值thresh，取A的起点A和终点B加入T；

step2、取S中的一点C，使之距离A和B连成的直线最远；

step3、如果距离大于阈值，则将C加入T；

step4、分别递归AC与CB；

step5、输出结果集T。

测试代码：

import cv2

mask = cv2.imread(r'C:\Users\Administrator\Desktop\aa.bmp', 0)  # read picture
mask[mask > 100] = 255
mask[mask != 255] = 0

im_show = cv2.cvtColor(mask, cv2.COLOR_GRAY2BGR)
countours, hierarchy = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
for idx, cnt in enumerate(countours):
    print("cnt shape:", cnt.shape) #　(N, 1, 2)
    rect = cv2.minAreaRect(cnt)  # (cx, cy), (bw, bh), theta
    # epsilon = max(min(bw, bh) * 0.1, 5)
    print("rect:", rect)
    epsilon = 0.1 * cv2.arcLength(cnt, True)  # 计算轮廓线长度
    print("epsilon:", epsilon)
    approx = cv2.approxPolyDP(cnt, epsilon, True)
    print("approx:", approx.shape)
    cv2.drawContours(im_show, approx, -1, (0, 0, 255), 5) # dtype=np.int32

    points = approx.reshape((-1, 2))
    if points.shape[0] < 4:
        continue
    print("points shape:", points.shape)
cv2.imwrite(r'C:\Users\Administrator\Desktop\im_show.jpg', im_show)

结果：

cv2.approxPolyDP函数实现轮廓线的多边形逼近