在用caffe自带的工具根据保存的log文件画相应的loss曲线,accuracy曲线时,按教程来总是出现各种各样的问题。在此记录一下亲测可用的方法。
1. 从gihub:https://github.com/CharleneZhao/caffe-plot-log 上下载下来这4个文件:
- extract_seconds.py
- parse_log.sh
- plot_learning_curve.py
- plot_training_log.py.example
2. 放在觉得方便的文件夹里,比如caffe/examples,并修改文件夹的权限
chmod -R 777 ./caffe/examples
3. 把生成的log文件同样放在一个文件夹里,例如train.log
cd caffe/examples
./plot_training_log.py.example 6 loss.png train.log
就好啦~同时文件夹里会生成 train.log.test 和 train.log.train 两个解析出来的文件 以及 画出来的图。
另外:
./plot_training_log.py.example 6 loss.png train.log
中的数字6 为不同的参数
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