您的位置:首页 > Web前端

caffe绘制loss,accuracy曲线

2017-06-23 09:01 281 查看
绘制loss曲线
 
²  必须先切换到gpu上(ssh gpu107),再使用caffe工具
1.    修改train.sh文件,添加训练日志log,再开始训练,如图所示:

LOG_log_dir=后面为log保存的位置

或者不添加训练日志,直接使用服务器生成的.err文件进行画图

2.    将caffe自带绘图工具拷贝到log所在的文件夹,相应的工具为:

caffe-master/tools/extra/parse_log.sh

caffe-master/tools/extra/extract_seconds.py

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

例如:cp caffe-master/tools/extra/parse_log.sh/home/lijun/caffe-master/examples/corn/Log

3.    在log所在文件夹执行命令

./ parse_log.sh  caffe.log

注意:若日志不是以.log结尾,则先需要重命名为.log文件

执行该命令后,会在当前文件夹下产生.train文件和.test文件,这两个文件即为所需数据,可用vi打开看看

4.   绘图

./plot_training_log.py.example0 save.png caffe.log

0代表曲线类型,save.png为图片名 caffe.log为对应日志

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


对于服务器终端界面无法显示图形,可将数据导出,用matlab绘图
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: