diff options
Diffstat (limited to 'megapixels/commands/datasets/lookup.py')
| -rw-r--r-- | megapixels/commands/datasets/lookup.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/megapixels/commands/datasets/lookup.py b/megapixels/commands/datasets/lookup.py index 5a2a171e..c1c66c19 100644 --- a/megapixels/commands/datasets/lookup.py +++ b/megapixels/commands/datasets/lookup.py @@ -13,7 +13,7 @@ log = Logger.getLogger() help='Vector index to lookup') @click.option('--data_store', 'opt_data_store', type=cfg.DataStoreVar, - default=click_utils.get_default(types.DataStore.NAS), + default=click_utils.get_default(types.DataStore.SSD), show_default=True, help=click_utils.show_help(types.Dataset)) @click.option('--dataset', 'opt_dataset', @@ -41,11 +41,12 @@ def cli(ctx, opt_index, opt_data_store, opt_dataset): log = Logger.getLogger() # init dataset dataset = Dataset(opt_data_store, opt_dataset) + #dataset.load_face_vectors() + dataset.load_records() + dataset.load_identities() # set data store and load files - dataset.load() # find image records - image_record = dataset.roi_idx_to_record(opt_index) - # debug + image_record = dataset.index_to_record(opt_index) image_record.summarize() # load image im = cv.imread(image_record.filepath) |