caffe绘制train和loss曲线

参考博文：caffe绘制训练过程的loss和accuracy曲线

1.必备文件

caffe/tools/extra下面有几个文件，必备的：

1 caffe/tools/extra/parse_log.sh  2 caffe/tools/extra/extract_seconds.py3 caffe/tools/extra/plot_training_log.py.example 

我在当前目录下新建了一个文件夹：log

把上面的三个文件爱呢拷贝到log下面：

然后把我训练生成的log文件：VGG_widerface_SSD_300x300_2.log文件拷贝到log文件下面

2.解析训练日志：

在当前目录下打开终端，执行：

./parse_log.sh VGG_widerface_SSD_300x300_2.log

此时，在log文件下面生成了VGG_widerface_SSD_300x300_2.log.train和VGG_widerface_SSD_300x300_2.log.test文件，里面保存了提取出来的seconds，loss，accuracy

信息。应该可以指定保存的路径，这里我偷懒直接都在当前目录log下面弄，自己到时候再拷贝是一样的。

3.绘制图片:

当前目录终端执行：

./plot_training_log.py.example 6 train_iters.png VGG_widerface_SSD_300x300_2.log

生成图片：

再执行命令：

./plot_training_log.py.example 7 train_loss_seconds.png VGG_widerface_SSD_300x300_2.log

得到：

说明：

因为我的总迭代次数非常大，并且设置的训练1w次时才开始验证，所以这里我只是中途看一下效果，看下loss变化情况

./plot_training_log.py.example 7 train_loss_seconds.png VGG_widerface_SSD_300x300_2.log

其中：

• 7 为曲线类型

• train_loss_seconds.png为保存图片的文件名

• train_loss_seconds为上述所得日志

参数的含义：

Supported chart types: 0: Test accuracy vs. Iters 1: Test accuracy vs. Seconds 2: Test loss vs. Iters 3: Test loss vs. Seconds 4: Train learning rate vs. Iters 5: Train learning rate vs. Seconds 6: Train loss vs. Iters 7: Train loss vs. Seconds