def set_up_parser()

in Damage Assessment Visualizer/utils/inference.py [0:0]

• 32 lines of code
• 1 McCabe index (conditional complexity)

def set_up_parser():
    """Set up the argument parser.
    Returns:
        the argument parser
    """
    parser = argparse.ArgumentParser(
        description=__doc__, formatter_class=argparse.ArgumentDefaultsHelpFormatter
    )

    parser.add_argument(
        "--pre-imagery",
        required=True,
        help="path to the pre imagery (RGB GeoTIFF in Byte format)",
        metavar="FILE",
    )
    parser.add_argument(
        "--post-imagery",
        required=True,
        help="path to the post imagery (RGB GeoTIFF in Byte format)",
        metavar="FILE",
    )
    parser.add_argument(
        "--output-fn",
        required=True,
        help="path to the GeoTIFF file to write output to",
        metavar="FILE",
    )
    parser.add_argument(
        "--gpu",
        default=0,
        type=int,
        help="GPU id to use for inference",
    )
    parser.add_argument(
        "-v", "--verbose", action="store_true", help="flag to print debugging to stdout"
    )

    return parser