我正在开发一个程序,我应该检测相同类型的形状,并用不同的颜色为每种类型着色。
我正在使用cv2.findCountours
然后cv2.approxPolyDP
检测每个形状。
该程序检测任何具有 8 条边的形状为圆形,因此我决定添加一些检查 - 我正在使用 检查当前轮廓的区域cv2.contourArea
,并且我也在检查cv2.minEnclosingCircle(cnt)
当前轮廓的区域。
如果它们相等,我们就有一个圆圈。
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, .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:
area = cv2.contourArea(cnt)
(cx, cy), radius = cv2.minEnclosingCircle(cnt)
circleArea = radius * radius * np.pi
print circleArea
print area
if circleArea == area:
cv2.drawContours(img, [cnt], 0, (220, 152, 91), -1)
cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()
我打印了每个区域,发现结果是不同的——即使形状明显是圆形。
例如,对于相同的形状,minEnclosureCircle 区域为 628.254637106,contourArea 为 569。另一个示例:minEnclosureCircle 区域为 2220.55512328,contourArea 为 2032.0。
我怎样才能正确计算这个面积?
我将不胜感激任何帮助!
我使用的图像:
和检测到的形状: