使用caffe绘制训练过程的loss和accuracy曲线

最近使用caffe训练时,不知道loss的变化趋势,于是就想到了能否将自己的训练数据图形化。后来经过查阅资料,发现caffe自带这样的工具:

caffe-master/tools/extra/parse_log.sh 
caffe-master/tools/extra/extract_seconds.py

caffe-master/tools/extra/plot_training_log.py.example 

使用这个方法的前提是存在log文件以及模型的.sh文件(具体怎么生成log文件还请自己百度)

1、解析训练日志

将最上面的3个脚本拷贝到Log文件夹下,执行:

./parse_log.sh   20180305.log

执行此代码,就会在当前文件夹下生成一个.train文件和.test文件

2、生成图片

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

这样就可以生成训练过程的各种数据曲线。0代表曲线的类型,下面有讲解。save为保存图片的名称,默认在当前路径。

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  
 


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值