VSCode编译python时如何设定参数和使用parser解析参数

1. Parameters Configuration

Open launch.json, then add the identify filter "args" to the file, such as follows, it has three paramters:

• --action "test"
• --batch_size "1"
• --weight "pytorchUnet/weights_0.pth"

{
    // Use IntelliSense to learn about possible attributes.
    // Hover to view descriptions of existing attributes.
    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
    "version": "0.2.0",
    "configurations": [
        {
            "name": "Python: 当前文件",
            "type": "python",
            "request": "launch",
            "program": "${file}",
            "console": "integratedTerminal",
            "args": [
                "--action", "test",
                "--batch_size", "1",
                "--weight", "pytorchUnet/weights_0.pth"
            ]
        }
    ]
}

2. How to Use Pramter Paser in Code

I dirctly paste the code of mine as an example:

if __name__ == '__main__':
    #参数解析
    parser = argparse.ArgumentParser() #创建一个ArgumentParser对象
    parser.add_argument('--action', type=str, help='train or test')#添加参数
    parser.add_argument('--batch_size', type=int, default=1)
    parser.add_argument('--weight', type=str, help='the path of the model weight file')
    args = parser.parse_args()
    
    if args.action == 'train':
        train()
    elif args.action == 'test':
        test()