python - 设置 GLOG_minloglevel=1 以防止从 Caffe 在 shell 中输出

Question

我正在使用 Caffe，它在加载神经网络时会向 shell 打印大量输出。
我想抑制该输出，据说可以通过GLOG_minloglevel=1在运行 Python 脚本时进行设置来完成。我尝试使用以下代码执行此操作，但我仍然从加载网络中获得所有输出。如何正确抑制输出？

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

score 48 · Accepted Answer

要抑制输出级别，您需要将 loglevel增加到至少 2

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

水平是

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

更新：
由于这个标志是全局的caffe，它必须在导入包之前设置（如jbumcaffe所指出的）。一旦设置并导入标志，GLOG 工具的行为就无法更改。caffe

score 28 · Accepted Answer

我能够让Shai 的解决方案正常工作，但只能在调用之前在 Python 中执行该行

import caffe

2 回答 2