OpenCV kmeans 錯誤：(-215) python

Question

錯誤：錯誤：(-215) (unsigned)labels[i] < (unsigned)K in function kmeans

self.cluster[i]表示一些計算出的像素位置。

img = numpy.asarray(img)
Z = img.reshape((-1,3))
Z = numpy.float32(Z)
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 10, 1.0)
#prepare centers
labels = numpy.zeros(k)
for i in range(k):
    labels[i] = self.cluster[i].j * img.shape[0] + self.cluster[i].i
print(Z)
ret, label, center = cv2.kmeans(Z, k, None, criteria, 10, cv2.KMEANS_USE_INITIAL_LABELS, labels)

我已閱讀有關 C++ 中相同錯誤的其他帖子，但我無法使其正常工作。 請幫忙！

編輯1：

import numpy as np
import sys
sys.path.append('/usr/local/lib/python3.5/site-packages')
import cv2

img = cv2.imread('image.jpg')
img = cv2.resize(img, (90,90))
Z = img.reshape((-1,3))

# convert to np.float32
Z = np.float32(Z)

# define criteria, number of clusters(K) and apply kmeans()
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 10, 1.0)
K = 4

labels = np.zeros(K)
labels[0] = 1
labels[1] = 100
labels[2] = 500
labels[3] = 1000
print(labels)
print(Z.shape)
ret,label,center=cv2.kmeans(Z,K,None,criteria,10,cv2.KMEANS_USE_INITIAL_LABELS, labels)

# Now convert back into uint8, and make original image
center = np.uint8(center)
res = center[label.flatten()]
res2 = res.reshape((img.shape))

cv2.imshow('res2',res2)
cv2.waitKey(0)
cv2.destroyAllWindows()

您可以單獨運行此代碼以檢查出了什么問題。 將圖像大小調整為小於 90x90 會導致它崩潰。 任何大於 91x91 的東西都可以工作。 誰能解釋為什么以及如何解決它？

Answer 1

使用cv2.KMEANS_PP_CENTERS或cv2.KMEANS_RANDOM_CENTERS而不是cv2.KMEANS_USE_INITIAL_LABELS時，我的問題解決了。