def from_image_dir(image_dir: Path = typer.Argument(..., exists=True, file_okay=False), with_box: bool = typer.Option(False, '--with-box/--no-box', '-wb/-nb'), output_coco: Path = typer.Option('/dev/stdout', '-o', dir_okay=False)): """ create empty coco file from image dir """ logger.debug('enter: from image dir') COCO.from_image_dir(image_dir=image_dir, with_box=with_box).to_json(output_coco, indent=consts.INDENT)
def visualize( coco_file: Path = typer.Argument(..., exists=True, dir_okay=False), img_dir: Path = typer.Argument(..., exists=True, file_okay=False), ): """ visualize inputs """ try: logger.debug('visualize {} with images in {}', coco_file, img_dir) COCO(coco_file).visualize(img_dir=img_dir) except (json.JSONDecodeError, TypeError): logger.info( f'non-coco input file {coco_file} detect, trying to convert') empty_gt = COCO.from_image_dir(img_dir).tmp_file_name() COCO.from_detect_file(str(coco_file), empty_gt).print_stat().visualize(img_dir=img_dir)