Function main

bench/python/run_bench.py:181–230

()

def main():
    # docs_tag: begin_parse_args
    parser = get_default_arg_parser(
        "Profiler for all ops of CV-CUDA.",
        input_path=os.path.join(current_dir, "assets", "brooklyn.jpg"),
        supports_video=False,
        batch_size=32,
    )
    parser.add_argument(
        "-n",
        "--num_iters",
        default=10,
        type=int,
        help="The number of iterations to run the benchmarks for.",
    )
    parser.add_argument(
        "--visualize",
        action="store_true",
        default=False,
        help="Flag specifying whether outputs from the operators should be visualized"
        " on written on disk or not.",
    )
    parser.add_argument(
        "ops",
        nargs="*",
        help="Optional list of one or more operator names which you want to benchmark. "
        "When supplied, the benchmarking will be restricted to only the operators that starts "
        "with these names.",
    )
    args = parse_validate_default_args(parser)

    logging.basicConfig(
        format="[%(name)s:%(lineno)d] %(asctime)s %(levelname)-6s %(message)s",
        level=getattr(logging, args.log_level.upper()),
        datefmt="%Y-%m-%d %H:%M:%S",
    )

    cvcuda_perf = CvCudaPerf("run_bench", default_args=args)
    run_bench(
        args.input_path,
        args.output_dir,
        args.batch_size,
        args.target_img_height,
        args.target_img_width,
        args.device_id,
        args.num_iters,
        args.visualize,
        args.ops,
        cvcuda_perf,
    )


if __name__ == "__main__":