- 在训练模型时,增加
-log_dir 路径
参数来保证在相应路径下保存所需绘图的日志文件。并且将所需要进行分析的日志文件的后缀改为.log
- 将
path_to_caffe\tools\extra
下的三个文件拷贝到当前工作目录下,需要拷贝的文件如下。
需要注意的是,可以将最后一个的文件名改为plot_training_log.py方便查看使用。并且如果想要生成完整的可视化图表,这三个文件缺一不可。 - 然后根据plot_training_log.py中的提示,在命令行中输入相应的参数,一般的命令格式为:
...\plot_training_log.py param_types where_you_store_image.png where_your_log_is.log
其中可供选择的param_number如下:
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
PS:在制作lmdb时候调用convert_imageset来进行数据转换
D:\new_caffe\caffe\build\tools>convert_imageset.exe D:\python_caffe_test\Classification/
D:\python_caffe_test\Classification\train.txt D:\python_caffe_test\Classification\train_lmdb -resize_width 224 -resize_height 224 --shuffle