首頁 > 軟體

Python argparse庫的基本使用步驟

2022-07-19 22:01:38

argparse庫是python下的一個命令列引數管理庫,支援int、str、float、bool、陣列等5種基本資料型別。在解析命令列引數時還支援預設值。此外,還支援修改命令列引數,在程式碼中增加命令列引數等操作。通過argparse庫,可以實現引數的簡易管理,控制程式碼執行中的一些流程或引數值。

基本用法

1、int支援

argparse的寫法 ,class_nums的預設值為7

parser.add_argument("--class_nums", type=int, default=7)

命令列傳參寫法(假設程式檔名稱為test.py), 傳入了新的值,覆蓋了預設值

test.py --class_nums 8

2、str支援

argparse的寫法,load_from沒有設定預設值,如果要設定的話請設定default

parser.add_argument("--load_from", type=str, help='load trained model to generate predictions of validation set')

命令列傳參寫法(假設程式檔名稱為test.py), 傳入了新的值"best.pth"賦值給load_from

test.py --load_from "best.pth"

3、float支援

argparse的寫法 ,lr的預設值為=0.0003

parser.add_argument("--lr", type=float, default=0.0003)

命令列傳參寫法(假設程式檔名稱為test.py), 傳入了新的值,覆蓋了預設值

test.py --lr 0.001

4、bool支援

argparse的寫法 ,pretrained的預設值為None。store_true表示傳入引數則為True

parser.add_argument("--pretrained", action="store_true")

命令列傳參寫法(假設程式檔名稱為test.py), 此時pretrained的值為True

test.py --pretrained

5、陣列支援

argparse的寫法 ,input_shape的預設值為[256, 256]。nargs表示傳入引數可能存在多個值,也就是說傳入的引數不一定是2個,3個也可以

parser.add_argument('--input_shape', nargs='+', type=int, default=[256, 256])

命令列傳參寫法(假設程式檔名稱為test.py), 此時input_shape的值為[512, 512]

test.py --input_shape 512 512

完整呼叫

以下程式碼實現了取引數值,修改引數值,新增引數值的完整呼叫。在非控制檯環境下執行下面程式碼是無法修改預設引數的,只能是在控制檯下執行才能修改引數

import argparse
class Options:
    def __init__(self):
        parser = argparse.ArgumentParser('名稱空間')
        parser.add_argument("--class_nums", type=int, default=7)
        parser.add_argument("--pretrained", action="store_true")
        parser.add_argument("--lr", type=float, default=0.0003)
        parser.add_argument("--load_from", type=str,
                            help='load trained model to generate predictions of validation set')
        parser.add_argument('--input_shape', nargs='+', type=int, default=[256, 256])
        self.parser = parser

    def parse(self):
        args = self.parser.parse_args()
        return args
if __name__ == "__main__":
    args = Options().parse()
    print("n原始引數:")
    print(args)
    
    print("n取引數值:")
    print("args.class_nums:",args.class_nums)
    
    print("n修改引數:")
    args.input_shape=[512,512]
    print(args)
    
    print("n新增引數:")
    args.new_key=[512,512]
    print(args)

博主測試程式碼:python .test.py --input_shape 128 128
執行輸出如下所示:

原始引數:
Namespace(class_nums=7, input_shape=[128, 128], load_from=None, lr=0.0003, pretrained=False)
取引數值:
args.class_nums: 7
修改引數:
Namespace(class_nums=7, input_shape=[512, 512], load_from=None, lr=0.0003, pretrained=False)
新增引數:
Namespace(class_nums=7, input_shape=[512, 512], load_from=None, lr=0.0003, new_key=[512, 512], pretrained=False)

到此這篇關於Python argparse庫的基本使用的文章就介紹到這了,更多相關Python argparse庫使用內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com