我正在使用Python和OpenCV来检测图像中的轮廓。但是,当我运行以下代码以使用轮廓索引仅绘制特定轮廓时,由于分配的索引是随机的,因此输出错误。
因此,我找到了质心(在我的情况下,所有质心都位于同一水平线上)。有什么方法可以根据质心的x值从左到右(从0到n)对轮廓索引进行排序?
您能否显示相同的代码?任何帮助将不胜感激!
contours, hierarchy = cv2.findContours(thresh,cv2.RETR_LIST,cv2.CHAIN_APPROX_SIMPLE)
i = 9 ## Contour Index <----
cv2.drawContours(img,contours[i:i+1],-1,(0,0,255),2)
centroids = []
for cnt in contours:
mom = cv2.moments(cnt)
(x,y) = int(mom['m10']/mom['m00']), int(mom['m01']/mom['m00'])
cv2.circle(org,(x,y),4,(255,255,255),-1)
centroids.append((x,y)
在python中实现列表的排序功能,例如here。
在已实现的功能中,您计算中心并验证X位置是否更大或更小。如果gretter返回1,则较小的-1等于0。
def greater(a, b):
momA = cv2.moments(a)
(xa,ya) = int(momA['m10']/momA['m00']), int(momA['m01']/momA['m00'])
momB = cv2.moments(b)
(xb,yb) = int(momB['m10']/momB['m00']), int(momB['m01']/momB['m00'])
if xa > xb:
return 1
if xa == xb:
return 0
else
return -1
如果只计算一次中心,可以肯定会做得更好。
然后就做
contours.sort(greater)