Caffe训练日志的可视化图表生成

2017-11-07  本文已影响0人  WZChan
  1. 在训练模型时,增加 -log_dir 路径参数来保证在相应路径下保存所需绘图的日志文件。并且将所需要进行分析的日志文件的后缀改为.log
    生成的日志文件,需要将后缀改为.log
    更改好的日志文件名格式
  2. path_to_caffe\tools\extra下的三个文件拷贝到当前工作目录下,需要拷贝的文件如下。
    extract_seconds.py
    parse_log.sh
    plot_training_log.py.example
    需要注意的是,可以将最后一个的文件名改为plot_training_log.py方便查看使用。并且如果想要生成完整的可视化图表,这三个文件缺一不可。
  3. 然后根据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

上一篇下一篇

猜你喜欢

热点阅读