caffe之工具

引言

最近跟小伙伴设计训练了很多模型,我们主要通过看mAP进行判断这个模型的好坏,没有将模型实际效果进行显示观察。这不,就想着写个调用的程序进行显示。

  • 显示训练过程中的loss变化情况
  • 显示训练过程中的lr变化情况
  • 模型的inference time
  • 模型效果的显示 本来想自己写,一看caffe里有类似的代码,真是太高兴了,先看看caffe是怎么做的。

Caffe Tools中的demo

caffe

caffe可执行文件可以有不同的选项进行选择功能,功能选择是通过功能函数指针注册的方式实现的,在tools/caffe.cpp中有,其中的注册功能部分大家有兴趣可以学习一下,这块还是很有趣的。

分析train log

caffe/tools/extra下有分析log的各种脚本,你可以使用gnuplot继续绘制,也可以采用python的matplot

  1. 如果想提取log的关键信息,可以查看parse_log.sh或者parse_log.py

  2. 如果想绘制采用绘制 python tools/extra/plot_training_log.py 2 examples/ooxx/result/result.png jobs/ResNet/VOC0712/OOXX_321x321/ResNet_VOC0712_OOXX_321x321.log

This script mainly serves as the basis of your customizations. Customization is a must. You can copy, paste, edit them in whatever way you want. Be warned that the fields in the training log may change in the future. You had better check the data files and change the mapping from field name to field index in create_field_index before designing your own plots.

Usage:
    ./plot_training_log.py chart_type[0-7] /where/to/save.png /path/to/fi
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值