使用openCV检测线的孔,末端和起点?

时间:2016-03-07 15:47:33

标签: python opencv tracking detect

我正在尝试创建一个Python脚本来检测一行的漏洞,结尾和开头。我认为openCV可以很好地实现这一点。

例如,所有内容都以此图片开头:

enter image description here

最后我想要实现的是:

enter image description here

所以我开始将图像导入Python并将其转换为灰度。现在我想到了使用goodFeaturesToTrack()方法跟踪洞的想法。它通常用于查找图像中的角落。

然而,这不能很好地工作,因为在那之后脚本知道点,但它不知道一个点是来自一个洞还是它是该行的开头或结尾。另一个问题是,如果我使用另一个图像,这个方法会检测到更多的点,而不仅仅是洞的开始和结束。

以下是我的完整代码,可以更好地理解我的问题:

import cv2
import numpy as np
import matplotlib.pyplot as plt

# lodes in img

img = cv2.imread('png1.png', cv2.IMREAD_COLOR)
img_gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)

corners = cv2.goodFeaturesToTrack(img_gray, 200, 0.05, 10)

for corner in corners:
    x, y = corner.ravel()
    cv2.circle(img, (x,y), 7, (255,255,0), -1)

cv2.imshow('img',img)

我不知道解决这个问题。

1 个答案:

答案 0 :(得分:2)

我添加了一个func getLandmarks()它返回所有的整体。所以在这里我假设如果半径为30像素的2个角落将被视为一个洞

if abs(x1-x2)<=30 and abs(y1-y2)<=30:

此行定义范围。

import cv2
import numpy as np

def getLandmarks(corners):
    holes=[]
    for i in range(0,len(corners)):
        for j in range(i+1,len(corners)):
            x1,y1=corners[i].ravel()
            x2,y2=corners[j].ravel()
            if abs(x1-x2)<=30 and abs(y1-y2)<=30:
                holes.append((int((x1+x2)/2),int((y1+y2)/2)))
    return holes

# lodes in img

img = cv2.imread('img.png', cv2.IMREAD_COLOR)
img_gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)

corners = cv2.goodFeaturesToTrack(img_gray, 200, 0.05, 10)

holes=getLandmarks(corners)
print len(holes)
for corner in holes:
    cv2.circle(img, (corner), 7, (255,255,0), -1)

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

输出

enter image description here

现在开始和结束您可以轻松地对X中的角(如果路径是从左到右)或Y(如果路径从上到下)进行排序,并且最小值和最大值将是您的开始和结束!