diff options
| author | Vasil Zlatanov <vasko@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal> | 2018-12-06 15:58:45 +0000 |
|---|---|---|
| committer | Vasil Zlatanov <vasko@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal> | 2018-12-06 15:58:45 +0000 |
| commit | 4edf5bc861517a8879b051106fd102474cafa518 (patch) | |
| tree | b12c28819f8b821f8f3a7d1880a0204af6ae52a6 /lib | |
| parent | 42d2cba4f2a9282f942b51c16e478ff2db096b3b (diff) | |
| download | vz215_np1915-4edf5bc861517a8879b051106fd102474cafa518.tar.gz vz215_np1915-4edf5bc861517a8879b051106fd102474cafa518.tar.bz2 vz215_np1915-4edf5bc861517a8879b051106fd102474cafa518.zip | |
Fix lazy range in python3
Diffstat (limited to 'lib')
| -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): |