在 Python 中从左到右对轮廓进行排序 (OpenCV)



我正在使用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中实现列表的排序功能,例如这里。

在实现的函数中,您可以计算中心并验证 X 位置是 gretter 还是更小。如果 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)

最新更新