1.准备工作
- 如果是远程登陆而不是在本机上跑,需要安装xshell和xming,才可以实现训练过程的可视化
- 参考这里
2.训练日志的保存
-
画图是根据训练日志中的数据来话的,所以需要保存训练日志
-
方法:
-
第一步,改训练命令脚本内容,添加产生日志的命令:
2>&1 | tee train.log
trian.log即为产生的日志文件,在caffe根目录下;
3.后续步骤,参考下面的链接第2步及以后;
- caffe中自带了这样的小工具 ,在tools/extra/下:
- caffe-master/tools/extra/parse_log.sh
- caffe-master/tools/extra/extract_seconds.py
- caffe-master/tools/extra/plot_training_log.py.example
将其拷贝至examples/myself/Log/下
4.提取训练日志中的accurcy和loss等信息
./parse_log.sh /xxx/train.log
这样就会在/xxx/文件夹下生成一个.train文件和一个.test文件
5.画图命令:
./plot_training_log.py.example 0 save.png trian.log
- 其中0代表测试精度和迭代次数的关系图,可以取代为:
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