python argparse的使用步驟(全網最全)

2023-04-08 06:00:38

一、前言

在學習深度學習會發現都比較愛用python這個argparse，雖然基本能理解，但沒有仔細自己動手去寫，因此這裡寫下來作為自己本人的學習筆記

argparse是python的一個命令列引數解析包，在程式碼需要頻繁修改引數時，方便使用，主要用法就是在命令列輸入自己想要修改的引數。

二、關於argparse用法常用框架

import argparse

def get_parser():

    # argparse.ArgumentParser生成argparse物件 description為描述資訊，當在命令列輸入需要顯示幫助資訊時，會顯示

    parser = argparse.ArgumentParser(description="pytorch unet training")

    # 路徑引數設定 help為引數的幫助資訊

    parser.add_argument("--data_path", default="./", help="DRIVE root")

    # 預測類別數量 type如果不指定需要輸入的是str型別

    parser.add_argument("--num_classes", default=1, type=int)

    # 指定裝置使用

    parser.add_argument("--device", default="cuda", help="training device")

    # 指定batch size大小 "-b", "--batch_size"表示兩個都可以在命令列使用

    parser.add_argument("-b", "--batch_size", default=4, type=int)

    return parser

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print(args)

如上get_parser()為本文會介紹的一些使用方法，這個python檔名為python_argparse_test1.py，其中

parser = argparse.ArgumentParser(description="pytorch unet training")用來建立parser物件

add_argument()用來增加引數

args = parser.parse_args()中parse_args()獲取解析的引數

1.獲取參數列

當命令列輸入python python_argparse_test1.py時，列印args獲取到的引數得到：

Namespace(batch_size=4, data_path='./', device='cuda', num_classes=1)

表示這個引數解析器解析到的參數列

2.獲取幫助資訊

輸入python python_argparse_test1.py -h或者python python_argparse_test1.py --help顯示資訊，其中usage顯示了它的用法，pytorch unet training為建立物件時的description,再下面是各個引數資訊與用法

3.命令列修改引數

import argparse

def get_parser():

    # argparse.ArgumentParser生成argparse物件 description為描述資訊，當在命令列輸入需要顯示幫助資訊時，會顯示

    parser = argparse.ArgumentParser(description="pytorch unet training")

    # 路徑引數設定 help為引數的幫助資訊 default為預設引數

    parser.add_argument("--data_path", default="./", help="DRIVE root")

    # 預測類別數量 type如果不指定需要輸入的是str型別

    parser.add_argument("--num_classes", default=1, type=int)

    # 指定裝置使用

    parser.add_argument("--device", default="cuda", help="training device")

    # 指定batch size大小 "-b", "--batch_size"表示兩個都可以在命令列使用

    parser.add_argument("-b", "--batch_size", default=4, type=int)

    return parser

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("data_path: ",args.data_path)

    print("num_classes: ", args.num_classes)

命令列輸入：python python_argparse_test1.py --data_path Desktop --num_classer 4，得到結果如下：

data_path: Desktop

num_classes: 4

可以看到通過命令列確實修改了引數

4.'_'與"__"的使用

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("-b: ",args.b)

    print("--batch_size: ", args.batch_size)

命令列輸入python python_argparse_test1.py -b 10 --batch_size 20，這時會報錯：

這是因為當'_'和'__'同時存在時，系統預設後者為引數名

把上述程式碼改為：

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("--batch_size: ",args.batch_size)

但命令列不受影響，繼續執行命令 python python_argparse_test1.py -b 10得到：

--batch_size: 10

5.type的使用

type會把輸入的命令列字元強制轉化為type的型別

if __name__ =='__main__':

    parser = get_parser()

    args = parser.parse_args()

    print("--batch_size type: ",type(args.batch_size))

命令列輸入：python python_argparse_test1.py --batch_size '10'得到：

--batch_size type: <class 'int'>

6.required:用來表示這個引數是否需要提供

parser.add_argument("--num_classes", default=1, type=int, required=True)

如果輸入命令python python_argparse_test1.py

這將會報錯提示確實需要的引數

python_argparse_test1.py: error: the following arguments are required: --num_classes

7.choices選擇引數

parser.add_argument('-arch', required=True, choices=['alexnet', 'vgg'])

如果執行命令：python python_argparse_test1.py -arch cnn

這將會報錯

python_argparse_test1.py: error: argument -arch: invalid choice: 'cnn' (choose from 'alexnet', 'vgg')

參考部落格：

argparse簡要用法總結 | Yunfeng's Simple Blog (vra.github.io)

argparse基本用法

argparse.ArgumentParser()用法解析

到此這篇關於python argparse的使用步驟(全網最全)的文章就介紹到這了,更多相關python argparse使用內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！