def parse_args()

in tools/convert_pkl_to_pb.py [0:0]

• 72 lines of code
• 4 McCabe index (conditional complexity)

def parse_args():
    parser = argparse.ArgumentParser(
        description="Convert a trained network to pb format"
    )
    parser.add_argument(
        "--cfg", dest="cfg_file", help="optional config file", default=None, type=str
    )
    parser.add_argument(
        "--net_name",
        dest="net_name",
        help="optional name for the net",
        default="detectron",
        type=str,
    )
    parser.add_argument(
        "--out_dir", dest="out_dir", help="output dir", default=None, type=str
    )
    parser.add_argument(
        "--test_img",
        dest="test_img",
        help="optional test image, used to verify the model conversion",
        default=None,
        type=str,
    )
    parser.add_argument(
        "--fuse_af", dest="fuse_af", help="1 to fuse_af", default=1, type=int
    )
    parser.add_argument(
        "--device",
        dest="device",
        help="Device to run the model on",
        choices=["cpu", "gpu"],
        default="cpu",
        type=str,
    )
    parser.add_argument(
        "--net_execution_type",
        dest="net_execution_type",
        help="caffe2 net execution type",
        choices=["simple", "dag"],
        default="simple",
        type=str,
    )
    parser.add_argument(
        "--use_nnpack",
        dest="use_nnpack",
        help="Use nnpack for conv",
        default=1,
        type=int,
    )
    parser.add_argument(
        "--logdb",
        dest="logdb",
        help="output to logfiledb instead of pb files",
        default=0,
        type=int,
    )
    parser.add_argument(
        "opts",
        help="See detectron/core/config.py for all options",
        default=None,
        nargs=argparse.REMAINDER,
    )
    if len(sys.argv) == 1:
        parser.print_help()
        sys.exit(1)
    ret = parser.parse_args()
    ret.out_dir = os.path.abspath(ret.out_dir)
    if ret.device == "gpu" and ret.use_nnpack:
        logger.warn("Should not use mobile engine for gpu model.")
        ret.use_nnpack = 0

    return ret