argparse模块使用说明

深度学习的工程中，py文件中的main函数一开始总会有大量的参数传入，而通常使用的方法就是使用argparse通过命令行传入，这篇博文旨在记录argparse的常用方法。

一、语句简介

1.载入argparse模块：

import argparse

2.创建语法分析器：

parser = argparse.ArgumentParser()

3.添加选项：

parser.add_argument("--epochs", type=int, default=100, help="number of epochs")
##
parser.add_argument("--", type=, default=, help="")

4.实例化语法分析器：

opt = parser.parse_args()

5.调用：

opt.epochs

二、标准使用方式

import argparse

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    # Super-Params
    parser.add_argument("--epochs", type=int, default=100, help="number of epochs")
    parser.add_argument("--batch_size", type=int, default=64, help="size of each image batch")
    parser.add_argument("--img_size", type=int, default=416, help="size of each image dimension")
    parser.add_argument("--n_cpu", type=int, default=8, help="number of cpu threads to use during batch generation")
    parser.add_argument("--gradient_accumulations", type=int, default=2, help="number of gradient accums before step")
    # Model config
    parser.add_argument("--model_def", type=str, default="config/yolov3-tiny_lsq.cfg", help="path to definition file")
    parser.add_argument("--pretrained_weights", type=str, help="if specified starts from checkpoint model")
    parser.add_argument("--gpu", type=int, default=1, help="assign a gpu to this porject, start from 0")
    # Eval & Save interval
    parser.add_argument("--checkpoint_interval", type=int, default=1, help="interval between saving model weights")
    parser.add_argument("--evaluation_interval", type=int, default=2, help="interval evaluations on validation set")
    # Dataset info
    parser.add_argument("--data_config", type=str, default="config/dac.data", help="path to data config file")
    # Others
    parser.add_argument("--compute_map", default=False, help="if True computes mAP every tenth batch")
    parser.add_argument("--multiscale_training", default=True, help="allow for multi-scale training")
    # Quantisize Settings:
    parser.add_argument("--quan_weitht_bit", type=int, default=8, help="")
    parser.add_argument("--quan_activation_bit", type=int, default=8, help="")

    opt = parser.parse_args()
    print(opt)