我想将红点连接在一起,以便每个红点仅连接一次到其最近的邻居红点。
答案 0 :(得分:0)
第一步,您应使用适当的工具(例如cv2.cvtColor(),cv2.threshold(),cv2.bitwise_not(),...)将图像转换为二进制图像-这意味着您的图片将仅包含黑色或白色像素。
示例:
然后,您应该在图像上找到轮廓(cv2.findContours),并使用尺寸标准(cv2.contourArea())将其过滤掉,以消除其他轮廓,例如图像中间的大五边形。
下一步,您应该找到每个轮廓的弯矩(cv2.moments()),这样您就可以获取轮廓中心的x和y坐标并将它们放在列表中。 (请小心地将右x和y坐标附加在一起)。
有了点之后,您可以计算所有点之间的距离(以及两个点之间的距离公式-sqrt((x2-x1)^ 2 +(y2-y1)^ 2))
然后,您可以使用任何逻辑来获取每个点最短距离的点的坐标(在下面的示例中,我将它们压缩在列表中,并创建了一个包含每个点的距离,x和y坐标的数组)。
代码示例:
import numpy as np
import cv2
img = cv2.imread('points.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret, threshold = cv2.threshold(gray,150,255,cv2.THRESH_BINARY)
cv2.bitwise_not(threshold, threshold)
im, contours, hierarchy = cv2.findContours(threshold,cv2.RETR_TREE,cv2.CHAIN_APPROX_NONE)
listx = []
listy=[]
for i in range(0, len(contours)):
c = contours[i]
size = cv2.contourArea(c)
if size < 1000:
M = cv2.moments(c)
cX = int(M["m10"] / M["m00"])
cY = int(M["m01"] / M["m00"])
listx.append(cX)
listy.append(cY)
listxy = list(zip(listx,listy))
listxy = np.array(listxy)
for i in range(0, len(listxy)):
x1 = listxy[i,0]
y1 = listxy[i,1]
distance = 0
secondx = []
secondy = []
dist_listappend = []
sort = []
for j in range(0, len(listxy)):
if i == j:
pass
else:
x2 = listxy[j,0]
y2 = listxy[j,1]
distance = np.sqrt((x1-x2)**2 + (y1-y2)**2)
secondx.append(x2)
secondy.append(y2)
dist_listappend.append(distance)
secondxy = list(zip(dist_listappend,secondx,secondy))
sort = sorted(secondxy, key=lambda second: second[0])
sort = np.array(sort)
cv2.line(img, (x1,y1), (int(sort[0,1]), int(sort[0,2])), (0,0,255), 2)
cv2.imshow('img', img)
cv2.imwrite('connected.png', img)
结果:
正如您在结果中看到的,现在每个点都与其最近的邻居点相连。希望它能有所帮助,或者至少对如何解决该问题有所帮助。干杯!