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)
