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

3
我正在使用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)
1个回答

3

实现python中类似于这里的列表排序函数。

在实现的函数中,您需要计算列表的中心,并验证X位置是大于还是小于其他位置。如果大于,则返回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)

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接