aboutsummaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/kmean.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/kmean.py b/lib/kmean.py
index f041b19..64ab166 100644
--- a/lib/kmean.py
+++ b/lib/kmean.py
@@ -21,7 +21,7 @@ def create_kmean_clusters(feature_vectors, labels, gallery_idx, camId):
for i in range(2):
km_train_data.append(KMeans(n_clusters=int(np.max(labels)),random_state=0).fit(train[i]))
km_idx.append(km_train_data[i].labels_)
- km_labels.append(range(np.max(labels)))
+ km_labels.append(list(range(np.max(labels))))
for j in range(np.max(labels)):
class_vote = np.zeros(np.max(labels))
for q in range(km_idx[i].size):