如何使用OpenCV将最近的点连接在一起

时间:2018-06-25 11:26:53

标签: opencv computer-vision

Red dots surrounding a pentagon

我想将红点连接在一起,以便每个红点仅连接一次到其最近的邻居红点。

1 个答案:

答案 0 :(得分:0)

第一步,您应使用适当的工具(例如cv2.cvtColor(),cv2.threshold(),cv2.bitwise_not(),...)将图像转换为二进制图像-这意味着您的图片将仅包含黑色或白色像素。

示例:

enter image description here

然后,您应该在图像上找到轮廓(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)

结果:

enter image description here

正如您在结果中看到的,现在每个点都与其最近的邻居点相连。希望它能有所帮助,或者至少对如何解决该问题有所帮助。干杯!