一. 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~~~
结束。