我正在使用 Caffe,它在加载神经网络时会向 shell 打印大量输出。
我想抑制该输出,据说可以通过在运行 Python 脚本时设置 GLOG_minloglevel=1
来完成。我尝试使用以下代码执行此操作,但我仍然从加载网络中获得所有输出。如何正确抑制输出?
os.environ["GLOG_minloglevel"] = "1"
net = caffe.Net(model_file, pretrained, caffe.TEST)
os.environ["GLOG_minloglevel"] = "0"
最佳答案
要抑制输出级别,您需要增加日志级别至少为 2
os.environ['GLOG_minloglevel'] = '2'
级别是
0 - 调试
1 - 信息(仍然有很多输出)
2 - 警告
3 - 错误
更新:
由于此标志是 global 到 caffe
,因此必须在 导入 caffe
包之前设置它(如所指出的)由 jbum 提供)。设置标志并导入 caffe
后,GLOG 工具的行为将无法更改。
关于python - 设置 GLOG_minloglevel=1 以防止在 shell 中从 Caffe 输出,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/29788075/