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)