我正在尝试创建一个Python脚本来检测一行的漏洞,结尾和开头。我认为openCV可以很好地实现这一点。
例如,所有内容都以此图片开头:
最后我想要实现的是:
所以我开始将图像导入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)
我不知道解决这个问题。
答案 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)
现在开始和结束您可以轻松地对X中的角(如果路径是从左到右)或Y(如果路径从上到下)进行排序,并且最小值和最大值将是您的开始和结束!