aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lib/kmean.py4
1 files changed, 0 insertions, 4 deletions
diff --git a/lib/kmean.py b/lib/kmean.py
index 2e34702..58309bc 100644
--- a/lib/kmean.py
+++ b/lib/kmean.py
@@ -34,15 +34,11 @@ def create_kmean_clusters(feature_vectors, labels, gallery_idx, camId):
clustercam = []
for i in range(2):
clustercam.append(np.add(np.ones(len(km_labels[i])), i))
- print(len(km_labels[i]))
for j in range(len(km_labels[i])):
cl.append(km_train_data[i].cluster_centers_[j])
cllab.append(km_labels[i][j])
train_data = np.array(cl)
- print(train_data.shape)
train_label = np.array(cllab)
- print(train_label.shape)
train_cam = np.array([clustercam[i] for i in range(2)]).reshape(train_label.shape[0])
- print(train_cam.shape)
return train_data, train_label, train_cam