aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorVasil Zlatanov <vasko@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-06 15:58:45 +0000
committerVasil Zlatanov <vasko@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-06 15:58:45 +0000
commit4edf5bc861517a8879b051106fd102474cafa518 (patch)
treeb12c28819f8b821f8f3a7d1880a0204af6ae52a6
parent42d2cba4f2a9282f942b51c16e478ff2db096b3b (diff)
downloadvz215_np1915-4edf5bc861517a8879b051106fd102474cafa518.tar.gz
vz215_np1915-4edf5bc861517a8879b051106fd102474cafa518.tar.bz2
vz215_np1915-4edf5bc861517a8879b051106fd102474cafa518.zip
Fix lazy range in python3
-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):