文章作者:Tyan
博客:noahsnail.com | CSDN | 简书
在训练Caffe模型后,部署Caffe服务时我们通常会使用pycaffe来加载模型并处理图像,但是pycaffe加载模型时通常会输出加载模型的日志,影响我们查看自己的日志,因此需要移除Caffe加载模型时的日志。代码如下:
os.environ['GLOG_minloglevel'] = '2'
import caffe
Caffe使用的日志是GLOG,其日志级别如下:
0 - debug
1 - info (still a LOT of outputs)
2 - warnings
3 - errors
注意:由于在导入Caffe时Caffe会加载GLOG,因此os.environ['GLOG_minloglevel'] = '2'
需要在import caffe
之前。
参考资料