diff options
Diffstat (limited to 'lib/kmean.py')
| -rw-r--r-- | lib/kmean.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/lib/kmean.py b/lib/kmean.py index 58309bc..f041b19 100644 --- a/lib/kmean.py +++ b/lib/kmean.py @@ -1,3 +1,4 @@ +from logging import debug import numpy as np from sklearn.cluster import KMeans @@ -39,6 +40,7 @@ def create_kmean_clusters(feature_vectors, labels, gallery_idx, camId): cllab.append(km_labels[i][j]) train_data = np.array(cl) + debug("Kmean data has shape", train_data.shape) train_label = np.array(cllab) train_cam = np.array([clustercam[i] for i in range(2)]).reshape(train_label.shape[0]) return train_data, train_label, train_cam |