参数-python的cv2.kmeans错误

时间:2016-11-12 21:25:29

标签: python opencv k-means

我对cv2有点新,我遇到以下错误

首先,我使用以下代码获取图像的ORB描述符:

import cv2
img = cv2.imread('messi.jpg',0) 
orb = cv2.ORB_create()
cv2.ocl.setUseOpenCL(False)
kp, des = orb.detectAndCompute(img, None)

该功能检测到500个关键点,每个关键点的维数为32.如果我这样做:

des.shape

我明白了:

(500L,32L) 

此外,我想使用kmeans获取描述符的质心。这是我的代码:

iterations = 10
epsilon = 1.0
k = 64
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, iterations, epsilon)
compactness, labels, centers = cv2.kmeans(des, k ,None ,criteria, iterations, cv2.KMEANS_RANDOM_CENTERS)

我得到的错误是:

 error: C:\builds\master_PackSlaveAddon-win64-vc12
 static\opencv\modules\core\src\kmeans.cpp:230: error: (-215) data0.dims <= 2
 && type == CV_32F && K > 0 in function cv::kmeans

我已经看到了C ++的这个错误的线程,它通常是图像维度或图像类型的问题,但在python中我不确定它意味着什么。

任何帮助?

1 个答案:

答案 0 :(得分:3)

错误的原因是cv2.kmeans函数的第一个参数必须是数据类型为32的数组

因此,引入以下代码行可以解决问题:

des = np.float32(des)

有关如何使用该功能的一些示例,请点击此处:

http://docs.opencv.org/3.0-beta/doc/py_tutorials/py_ml/py_kmeans/py_kmeans_opencv/py_kmeans_opencv.html