diff options
| author | nunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal> | 2018-12-10 02:42:54 +0000 |
|---|---|---|
| committer | nunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal> | 2018-12-10 02:42:54 +0000 |
| commit | 743ac705dac81896f30227fb949c3175ef751427 (patch) | |
| tree | ad3664a1042a5156bee3b7336c788e85e8de3818 /evaluate.py | |
| parent | a1985d4f28fdf596f4c223b34598f9d82eb5a154 (diff) | |
| download | vz215_np1915-743ac705dac81896f30227fb949c3175ef751427.tar.gz vz215_np1915-743ac705dac81896f30227fb949c3175ef751427.tar.bz2 vz215_np1915-743ac705dac81896f30227fb949c3175ef751427.zip | |
Spaces
Diffstat (limited to 'evaluate.py')
| -rwxr-xr-x | evaluate.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/evaluate.py b/evaluate.py index 9218649..9910360 100755 --- a/evaluate.py +++ b/evaluate.py @@ -56,9 +56,8 @@ parser.add_argument("-V", "--validation", help="Validation Mode", action='store_ parser.add_argument("-K", "--kmean_alt", help="Kmean", type=int, default=0) parser.add_argument("-P", "--mAP", help="Mean Average Precision", action='store_true') - - args = parser.parse_args() + if args.verbose: logging.basicConfig(level=logging.DEBUG) @@ -223,7 +222,7 @@ def main(): if(args.kmean): debug("Using Kmeans") train_data, train_label, train_cam = create_kmean_clusters(feature_vectors, labels,gallery_idx,camId) - + if args.kmean_alt: kmeans = KMeans(n_clusters=args.kmean_alt, random_state=0).fit(train_data) neigh = NearestNeighbors(n_neighbors=1) |