美文网首页Caffe
Caffe关闭日志输出

Caffe关闭日志输出

作者: SnailTyan | 来源:发表于2017-07-18 13:41 被阅读60次

    文章作者: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之前。

    参考资料

    1. https://stackoverflow.com/questions/29788075/setting-glog-minloglevel-1-to-prevent-output-in-shell-from-caffe

    相关文章

      网友评论

        本文标题:Caffe关闭日志输出

        本文链接:https://www.haomeiwen.com/subject/pffhkxtx.html