def main()

in mapillary_vistas/tools/evaluation.py [0:0]

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

def main():
    parser = argparse.ArgumentParser(description="Commandline tool for evaluating predictions", formatter_class=RawTextHelpFormatter)
    parser.add_argument(
        '-c',
        '--dataset-config',
        type=argparse.FileType(),
        action='store',
        required=True,
        help="config.json of the dataset. Used for label names")

    parser.add_argument(
        '-g',
        '--ground-truth-labels',
        type=str,
        action='store',
        required=True,
        help="Path to the folder with png ground truth label data.")

    parser.add_argument(
        '-p',
        '--prediction-labels',
        type=str,
        action='store',
        default=None,
        help="Path to the folder with predicted label data. "
             "(PNG files, file name must be exactly the same as ground truth data)")

    parser.add_argument(
        '-i',
        '--instances',
        type=str,
        action='store',
        default=None,
        help="""Path to the folder with instance description.