我有两个点阵列,我也知道它们在我们世界中的位置。我需要计算每个数组的总中间以及这个中间的2d(xy)坐标。因此,应该输入具有这些点阵列的图像,并且输出应该是2个列表:return[x1,y1],[x2,y2]。Python3.x语言,OpenCV库
我发现了一些有用的代码(Python):
import numpy as np
import cv2 as cv
from matplotlib import pyplot as plt
X = np.random.randint(25,50,(5,2))
Y = np.random.randint(60,85,(4,2))
Z = np.vstack((X,Y))
# convert to np.float32
Z = np.float32(Z)
# define criteria and apply kmeans()
criteria = (cv.TERM_CRITERIA_EPS + cv.TERM_CRITERIA_MAX_ITER, 10, 1.0)
ret,label,center=cv.kmeans(Z,2,None,criteria,10,cv.KMEANS_RANDOM_CENTERS)
# Now separate the data, Note the flatten()
A = Z[label.ravel()==0]
B = Z[label.ravel()==1]
# Plot the data
plt.scatter(A[:,0],A[:,1])
plt.scatter(B[:,0],B[:,1],c = 'r')
plt.scatter(center[:,0],center[:,1],s = 80,c = 'y', marker = 's')
plt.xlabel('Height'),plt.ylabel('Weight')
plt.show()
此代码在点数组中查找中点。 但问题是理解哪个点属于哪个数组
一个想法是阈值,执行变形操作,找到轮廓,然后找到质心。这应该给你两点
变形关闭以连接点
以蓝色绘制质心的结果
坐标
代码
相关问题 更多 >
编程相关推荐