示例#1
0
@click.option('--dry-run',
              'opt_dry_run',
              is_flag=True,
              default=False,
              show_default=True)
@click.option('--root',
              'opt_dir_cli_root',
              default=app_cfg.DIR_CLI,
              help='Path to root directory of vframe_synthetic code')
@click.option('--system',
              'opt_system',
              type=types.GeneratorSystemVar,
              default=click_utils.get_default(types.GeneratorSystem.STATIC),
              required=True,
              show_default=True,
              help=click_utils.show_help(types.GeneratorSystem))
@click.pass_context
def cli(ctx, opt_fp_blender, opt_fp_blend, opt_system, opt_fp_cfg,
        opt_checkpoint, opt_dry_run, opt_dir_cli_root):
    """Runs Blender synthetic data generator"""

    import subprocess

    log = app_cfg.LOG
    log.info('Running Blender generator')

    # Blender variables. The trailing '--' indicates end of arguments
    fn_generator = f'{opt_system.name.lower()}.py'
    fp_generator = join(app_cfg.DIR_GENERATORS, fn_generator)
    args = [
        opt_fp_blender, opt_fp_blend, '--background', '--python', fp_generator,
示例#2
0
"""Template script
"""

import click

from app.models import types
from app.utils import click_utils


@click.command('')
@click.option('-m',
              '--model',
              'opt_model_enum',
              type=types.ModelZooClickVar,
              required=True,
              help=click_utils.show_help(types.ModelZoo))
@click.pass_context
def cli(ctx, opt_model_enum):
    """Print DNN layer info to find feat vec layer"""

    # ------------------------------------------------
    # imports

    from os.path import join
    import logging
    from urllib import request
    from pathlib import Path

    import numpy as np

    from app.image import cvdnn