diff options
| -rw-r--r-- | lib/kmean.py | 2 |
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): |