From 5340bee951c18910fd764241945f1f136b5a22b4 Mon Sep 17 00:00:00 2001 From: adamhrv Date: Sun, 23 Dec 2018 01:24:24 +0100 Subject: fixing face roi --- megapixels/commands/demo/face_search.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'megapixels/commands/demo') diff --git a/megapixels/commands/demo/face_search.py b/megapixels/commands/demo/face_search.py index 34a25762..6e4bcdad 100644 --- a/megapixels/commands/demo/face_search.py +++ b/megapixels/commands/demo/face_search.py @@ -21,6 +21,8 @@ log = Logger.getLogger() required=True, show_default=True, help=click_utils.show_help(types.Dataset)) +@click.option('--results', 'opt_results', default=5, + help='Number of match results to display') @click.option('--gpu', 'opt_gpu', default=0, help='GPU index (use -1 for CPU') @click.pass_context @@ -73,7 +75,7 @@ def cli(ctx, opt_fp_in, opt_data_store, opt_dataset, opt_gpu): vec_query = recognition.vec(im_query, bbox) # find matches - image_records = dataset.find_matches(vec_query, n_results=5) + image_records = dataset.find_matches(vec_query, n_results=opt_results) # summary ims_match = [im_query] -- cgit v1.2.3-70-g09d2