设置GLOG_minloglevel=1以防止shell中的输出来自C

2024-04-29 08:59:32 发布

您现在位置:Python中文网/ 问答频道 /正文

我使用的是Caffe,它在加载神经网络时会将大量输出打印到shell中。
我想禁止该输出,这可以通过在运行Python脚本时设置GLOG_minloglevel=1来完成。我已经试过用下面的代码来做了,但是我仍然可以从加载网络中得到所有的输出。如何正确抑制输出?

os.environ["GLOG_minloglevel"] = "1"
net = caffe.Net(model_file, pretrained, caffe.TEST)
os.environ["GLOG_minloglevel"] = "0"

Tags: 代码网络脚本netmodelosenviron神经网络
2条回答

我能够让Shai's solution工作,但只有在调用

import caffe

要控制输出级别,您需要将日志级别至少增加到2

 os.environ['GLOG_minloglevel'] = '2' 

级别是

0-调试
1-信息(仍然有很多输出)
2-警告
3-错误


更新:
由于此标志是globalcaffe,因此必须在导入caffe包之前设置(如jbum所指出)。一旦设置了标志并导入了caffe,就不能更改GLOG工具的行为。

相关问题 更多 >