caffe 中保存log到文本

修改caffe-master\src\caffe文件夹下的common.cpp文件,或者用vs打开caffe解决方案。

编译common.cpp文件。

添加头文件

1

2

#include <boost/date_time.hpp> 

#include <direct.h> 

找到GlobalInit函数,向其中添加代码

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

void GlobalInit(int* pargc, char*** pargv) {

  // Google flags.

  ::gflags::ParseCommandLineFlags(pargc, pargv, true);

  // Google logging.

   

    std::string FLAGS_log_dir="./log/";//存放日志文件的文件夹路径,我们可以自己指定 

    _mkdir(FLAGS_log_dir.c_str());

    FLAGS_colorlogtostderr = true;//设置输出到屏幕的日志显示相应颜色

    std::string LOG_INFO_FILE; 

    std::string LOG_WARNING_FILE; 

    std::string LOG_ERROR_FILE; 

    std::string LOG_FATAL_FILE; 

    std::string now_time = boost::posix_time::to_iso_extended_string(boost::posix_time::second_clock::local_time()); 

    now_time[13] = '-'

    now_time[16] = '-'

    LOG_INFO_FILE = FLAGS_log_dir + "INFO" + now_time + ".txt"

    google::SetLogDestination(google::GLOG_INFO, LOG_INFO_FILE.c_str()); 

    LOG_WARNING_FILE = FLAGS_log_dir + "WARNING" + now_time + ".txt"

    google::SetLogDestination(google::GLOG_WARNING, LOG_WARNING_FILE.c_str()); 

    LOG_ERROR_FILE = FLAGS_log_dir + "ERROR" + now_time + ".txt"

    google::SetLogDestination(google::GLOG_ERROR, LOG_ERROR_FILE.c_str()); 

    LOG_FATAL_FILE = FLAGS_log_dir + "FATAL" + now_time + ".txt"

    google::SetLogDestination(google::GLOG_FATAL, LOG_FATAL_FILE.c_str()); 

 

  ::google::InitGoogleLogging(*(pargv)[0]);

  // Provide a backtrace on segfault.

 

#if !defined(_MSC_VER)

  ::google::InstallFailureSignalHandler();

#endif

}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值