关于python代码中的sys.argv[]作用
在学习tensorflow时,经常出现直接在cmd上使用脚本语言进行模型训练工作,这使得整个过程很不透明也不容易使用debug。在我们观察源码时往往会看到sys.argv[]
这一语句,下面以人脸识别时的train_softmax.py文件为例。
代码主函数为
if __name__ == '__main__':
main(parse_arguments(sys.argv[1:]))
而在运行时cmd窗口处代码为
python train_softmax.py \
--log_base_dir ~/logs/myface/ \
--model_base_dir ~/models/myface/ \
--data_dir ~/datasets/casia/casia_maxpy_mtcnnpy_182 \
...
于是我们猜想sys.argv[1:]
其实起到一个读入参数并传参的功能, 他实现将读入的字符串从第
位到最后一位全部传入。
为了验证这一猜想我们编写一个测试代码test_sys_argv.py
代码内容如下:
import sys
a=sys.argv[1:]
print(a)
我们在命令窗口输入
python test_sys_argv.py \
~/logs/myface/ \
~/models/myface/ \
~/datasets/casia/casia_maxpy_mtcnnpy_182
得到
而当我们将test_sys_argv.py改为
import sys
a=sys.argv[0:]
print(a)
得到
于是我们的猜想得到验证