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