argparse模块

Posted 2020-09-21 去做点事情

模块argparse是python标准库中推荐的命令行解析模块

简明使用步骤：

1：import argparse　　　　　　　　　　　　　　　　　　首先导入该模块

2：parser = argparse.ArgumentParser()　　　　　　然后创建一个解析对象

3：parser.add_argument()　　　　　　　　　　　　　　然后向该对象中添加你要关注的命令行参数和选项，每一个add_argument方法对应一个你要关注的参数或选项

4：parser.parse_args()　　　　　　　　　　　　　　　最后调用parse_args()方法进行解析，解析成功之后即可使用

以下是fast-rcnn代码中train_net的使用：

def parse_args():
    """
    Parse input arguments
    """
    parser = argparse.ArgumentParser(description=\'Train a Fast R-CNN network\')
    parser.add_argument(\'--gpu\', dest=\'gpu_id\',
                        help=\'GPU device id to use [0]\',
                        default=0, type=int)
    parser.add_argument(\'--solver\', dest=\'solver\',
                        help=\'solver prototxt\',
                        default=None, type=str)
    parser.add_argument(\'--iters\', dest=\'max_iters\',
                        help=\'number of iterations to train\',
                        default=40000, type=int)
    parser.add_argument(\'--weights\', dest=\'pretrained_model\',
                        help=\'initialize with pretrained model weights\',
                        default=None, type=str)
    parser.add_argument(\'--cfg\', dest=\'cfg_file\',
                        help=\'optional config file\',
                        default=None, type=str)
    parser.add_argument(\'--imdb\', dest=\'imdb_name\',
                        help=\'dataset to train on\',
                        default=\'voc_2007_trainval\', type=str)
    parser.add_argument(\'--rand\', dest=\'randomize\',
                        help=\'randomize (do not use a fixed seed)\',
                        action=\'store_true\')
    parser.add_argument(\'--set\', dest=\'set_cfgs\',
                        help=\'set config keys\', default=None,
                        nargs=argparse.REMAINDER)

    if len(sys.argv) == 1:
        parser.print_help()
        sys.exit(1)

    args = parser.parse_args()
    return args

这是没加任何参数直接输入的情况：

这和输入python train_net.py -h是一样的效果，因为原代码中有这样一段代码：

if len(sys.argv) == 1:
        parser.print_help()
        sys.exit(1)