我使用的是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:
我能够让Shai's solution工作,但只有在调用
要控制输出级别,您需要将日志级别至少增加到2
级别是
0-调试
1-信息(仍然有很多输出)
2-警告
3-错误
更新:
由于此标志是global到
caffe
,因此必须在导入caffe
包之前设置(如jbum所指出)。一旦设置了标志并导入了caffe
,就不能更改GLOG工具的行为。相关问题 更多 >
编程相关推荐