diff options
Diffstat (limited to 'megapixels/commands/datasets/lookup.py')
| -rw-r--r-- | megapixels/commands/datasets/lookup.py | 26 |
1 files changed, 11 insertions, 15 deletions
diff --git a/megapixels/commands/datasets/lookup.py b/megapixels/commands/datasets/lookup.py index e84bdf3e..5a2a171e 100644 --- a/megapixels/commands/datasets/lookup.py +++ b/megapixels/commands/datasets/lookup.py @@ -6,8 +6,10 @@ from app.utils import click_utils from app.settings import app_cfg as cfg from app.utils.logger_utils import Logger +log = Logger.getLogger() + @click.command() -@click.option('--index', 'opt_index', type=int, +@click.option('--index', 'opt_index', type=int, required=True, help='Vector index to lookup') @click.option('--data_store', 'opt_data_store', type=cfg.DataStoreVar, @@ -19,12 +21,8 @@ from app.utils.logger_utils import Logger required=True, show_default=True, help=click_utils.show_help(types.Dataset)) -@click.option('--metadata', 'opt_metadata_type', required=True, - type=cfg.MetadataVar, - show_default=True, - help=click_utils.show_help(types.Metadata)) @click.pass_context -def cli(ctx, opt_index, opt_data_store, opt_dataset, opt_metadata_type): +def cli(ctx, opt_index, opt_data_store, opt_dataset): """Display image info""" import sys @@ -37,22 +35,20 @@ def cli(ctx, opt_index, opt_data_store, opt_dataset, opt_metadata_type): import cv2 as cv from tqdm import tqdm - from app.utils import file_utils, im_utils, path_utils + from app.utils import file_utils, im_utils + from app.models.data_store import DataStore log = Logger.getLogger() - - log.info(f'creating dataset: {opt_dataset}') - dataset = Dataset(opt_dataset) - # loads all CSV files, may take a while - log.info(f'loading dataset...') - dataset.load(opt_data_store) + # init dataset + dataset = Dataset(opt_data_store, opt_dataset) + # set data store and load files + dataset.load() # find image records image_record = dataset.roi_idx_to_record(opt_index) # debug image_record.summarize() # load image - fp_im = image_record.filepath - im = cv.imread(fp_im) + im = cv.imread(image_record.filepath) # display cv.imshow('', im) # cv gui |