Python argparse庫的基本使用步驟

Posted on by

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庫使用內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：