您的位置:首页 > Web前端

caffe使用自带的工具绘制accuracy和loss曲线

2016-11-30 17:17 661 查看
caffe中其实已经自带了这样的小工具 caffe-master/tools/extra/parse_log.sh caffe-master/tools/extra/extract_seconds.py和 caffe-master/tools/extra/plot_training_log.py.example ,使用方法如下:

1.在训练的时候生成log文件。其中,log文件名必须是小写的。训练脚本可以按照下面的格式来写。

$TOOLS/caffe train --solver=$SOLVERFILE 2>&1 |tee out.log


2.生成图片

./plot_training_log.py.example 0  save.png out.log


就可以生成训练过程中的Test accuracy vs. Iters 曲线,其中0代表曲线类型, save.png 代表保存的图片名称, save.png 存放路径位于plot_training_log.py.example 所在路径下。

caffe中支持很多种曲线绘制,通过指定不同的类型参数即可,具体参数如下:

Notes:
1. Supporting multiple logs.
2. Log file name must end with the lower-cased ".log".
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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  caffe