caffe2 mdl文件转init_net.pb, predict_net.pb
来源:互联网 发布:农村淘宝一单赚多少钱 编辑:程序博客网 时间:2024/06/18 07:57
前言
caffe2中,完整的训练例子就train_resnet50.py, 其模型和snapshot都保持在同一个文件中,不像caffe 一样有.caffemodel 和 .snapshot 文件(其实是冗余的)。训练完之后发现*.mdl文件不能直接使用,需要转化为两个pb文件,然而没有说怎么转换,也有很多人在问这个issue,然而并没有人说出正确的方法。
解决方法
通过看代码(好像也没有别的方法了),发现了可以通过
from caffe2.python.predictor.mobile_exporter import Export
这个文件函数来进行转换。高高兴兴的转换了,一测试,又报错说啥*-rm *_riv没有定义啥的,也就是说关于BN的运行均值和方差没有存入到网络中,需要手动指定一下(没有BN的网络没有这个问题),简单的包装一下,如下:
from caffe2.python import core, model_helper, net_drawer, workspace, visualize, brewfrom caffe2.python.predictor.mobile_exporter import Exportdef export(workspace, net, params, init_net_name="init_net.pb", predict_net_name="predict_net.pb" ): extra_params = [] extra_blobs = [] for blob in workspace.Blobs(): | name = str(blob) | if name.endswith("_rm") or name.endswith("_riv"): | | extra_params.append(name) | | extra_blobs.append(workspace.FetchBlob(name)) for name, blob in zip(extra_params, extra_blobs): | workspace.FeedBlob(name, blob) | params.append(name) init_net, predict_net = Export(workspace, net, params) with open(init_net_name, 'wb') as f: | f.write(init_net.SerializeToString()) with open(predict_net_name, 'wb') as f: | f.write(predict_net.SerializeToString())
这样,只需要在需要的保存的时候,调用
export(workspace, deploy_model.net, deploy_model.params)
就可以了。
阅读全文
0 0
- caffe2 mdl文件转init_net.pb, predict_net.pb
- PB
- pb
- PB
- PB
- PB
- pb
- PB如何读取文件
- PB如何读取文件
- PB导入EXCEL文件
- PB读取EXCEL文件
- pb实现文件copy
- pb调用DLL文件
- 生成pb文件详解
- pb文件的使用
- 生成pb文件详解
- [PB]pb转excel 批注的操作
- pb中的打印----转
- Linux 常用命令-目录文件管理二
- c++ 非托管程序调用c#托管程序l的实现操作方式
- CentOS 7 SSH 免密登录的方法
- 数据结构 -- 栈的数组实现法
- CentOS 7 设置默认进入图形界面或文本界面
- caffe2 mdl文件转init_net.pb, predict_net.pb
- 梯度下降优化算法综述
- 引用第三方进行Android前端与web后台的数据交互
- vue-lazyload基础实例(基于vue2.0和vue-router2.0)
- 数字电路中的亚稳态产生原因和处理方法
- 修饰符
- Python-14 函数:灵活即强大
- IO之流输入输出FileInputStream和FileOutputStream
- Bigdata Development Java_Study_04(interface 堆,栈)