Python argparse库的基本使用步骤

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

test.py --class_nums 8

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

test.py --load_from "best.pth"

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

test.py --lr 0.001

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

test.py --pretrained

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

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)