-1

我正在开发一个程序，我应该检测形状——圆形、正方形和三角形——并用不同的颜色为每种类型着色。

我使用 cv2.findCountours 然后 cv2.approxPolyDP 来检测每个形状。

这是我的代码：

import numpy as np
import cv2

img = cv2.imread('1.jpg')
gray = cv2.imread('1.jpg',0)

ret,thresh = cv2.threshold(gray,127,255,1)

contours,h = cv2.findContours(thresh,cv2.RETR_CCOMP,cv2.CHAIN_APPROX_NONE)

for cnt in contours:
    approx = cv2.approxPolyDP(cnt, 0.03 * cv2.arcLength(cnt, True), True)
    print len(approx)
    if len(approx)==3:
        print "triangle"
        cv2.drawContours(img,[cnt],0,(122,212,78),-1)
    elif len(approx)==4:
        print "square"
        cv2.drawContours(img,[cnt],0,(94,234,255),-1)
    elif len(approx) == 8:
        k = cv2.isContourConvex(approx)
        if k:
            cv2.drawContours(img, [cnt], 0, (220, 152, 91), -1)


cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()

它确实检测到我提到的形状，但它也检测到不是圆形/三角形/正方形的形状并按原样呈现它们。

这是我使用的图像：1

输出：2

任何建议如何解决这个问题？我可以添加哪些考试？

谢谢！

4

1 回答 1

1

我看到您按照我的建议修改了代码，如果凸性检查解决了您的问题，请关闭上一个问题。对于答案：

对不同的形状使用不同的 epsilon 值。您的代码中的 epsilon 值为 0.03

approx = cv2.approxPolyDP(cnt, 0.03 *   cv2.arcLength(cnt, True), True)

但它可能会随轮廓的分辨率和面积而变化，因此您应该为不同的形状尝试不同的 epsilon 值

于 2017-10-09T12:31:17.097 回答