summaryrefslogtreecommitdiff
path: root/megapixels/commands/demo/face_search.py
diff options
context:
space:
mode:
Diffstat (limited to 'megapixels/commands/demo/face_search.py')
-rw-r--r--megapixels/commands/demo/face_search.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/megapixels/commands/demo/face_search.py b/megapixels/commands/demo/face_search.py
index 6e4bcdad..ca0b8016 100644
--- a/megapixels/commands/demo/face_search.py
+++ b/megapixels/commands/demo/face_search.py
@@ -15,7 +15,7 @@ log = Logger.getLogger()
type=cfg.DataStoreVar,
default=click_utils.get_default(types.DataStore.SSD),
show_default=True,
- help=click_utils.show_help(types.Dataset))
+ help=click_utils.show_help(types.DataStore))
@click.option('--dataset', 'opt_dataset',
type=cfg.DatasetVar,
required=True,
@@ -26,7 +26,7 @@ log = Logger.getLogger()
@click.option('--gpu', 'opt_gpu', default=0,
help='GPU index (use -1 for CPU')
@click.pass_context
-def cli(ctx, opt_fp_in, opt_data_store, opt_dataset, opt_gpu):
+def cli(ctx, opt_fp_in, opt_data_store, opt_dataset, opt_results, opt_gpu):
"""Display image info"""
import sys