python命令行参数：sys、argparse模块

一. sys

假设有一个函数文件为test.py，内容为：

import sys
a = sys.argv[0]
b = sys.argv[1]
print(a)  # test.py
print(b)  # 1000

• 运行python test.py 1000，可以发现sys.argv[0]就是test.py本身，sys.argv[1]就是传入的第一个命令行参数1000，注意这里的1000是str类型。

• 如果命令输入多个参数（按空格隔开），会把多个参数存在sys.argv里，实际上sys.argv就是个列表类型，用来存储命令行参数。例子如下：

import sys
print(type(sys.argv))  # list
c = sys.argv[1:]
print(c) 

运行python test.py 1 2 3 4，返回结果如下图所示：

二. argparse

argparser模块经常会在训练模型的时候看到，例如gpu_ids的选择，batch_size，epochs等等，argparser这个模块也可以传入命令行参数。具体用法如下所述：

• 首先导入模块，创建一个解析器:

import argparse
parser = argparse.ArgumentParser(description='description')
# description后面是描述信息。

• 添加参数，如下所示添加了三种类型的参数，设置了默认值，表示如果不从命令行传入，就使用默认值 。

parser.add_argument('--gpu_ids',default='0', type=str, help='gpu_ids: e.g. 0  0,1,2  0,2')
parser.add_argument('--batchsize', default=256, type=int, help='batchsize')
parser.add_argument('--use_dense', action='store_true', help='use densenet121' )

• 解析参数

args = parser.parse_args() 

• 打印结果，看一下参数：

# 不传入命令行参数
print(args)
print(args.gpu_ids)
print(args.batchsize)
print(args.use_dense)

不传入参数，则打印出默认参数。

传入命令行参数，运行python test.py --batchsize 64 --use_dense 则有

三. 总结

没什么好总结的，就是记录一下，hiahiahia~~~

结束。