caffe转换tensorflow工具caffe-tensorflow

caffe-tensorflow工具能够将caffe模型转换为tensorflow格式,包括网络结构源码和权重文件。通过运行convert.py,输入参数包括caffe的网络定义文件和预训练参数,输出为tensorflow代码和npy权重。转换过程中解决了卷积核、全连接层权重、数据格式等不匹配问题。
摘要由CSDN通过智能技术生成

caffe-tensorflow转换介绍及使用

caffe-tensorflow可以将caffe的网络定义文件和预训练参数转换成tensorflow的形式,包括tensorflow的网络结构源码和npy格式的权重文件。
从github上下载源码,进入源码目录运行convert.py即可。
其调用格式为

python convert.py def_path --caffemodel caffemodel_path --data-output-path dataoutput_path --code-output-path codeoutput_path

其中def_path为必需项,四个参数两个caffe输入,两个tensorflow端的输出。如果只需要权重参数则只需前三个参数。输出的tensorflow代码为其源码中Network的子类,网络构建过程实现了其setup方法,因此需在代码中引入才可以调用。tensorflow目录下network.py这个文件中实现了Network类,这个类封装了一些layer,包括带group的conv层,caffe-tensorflow的转换还是比较安全的,自行转换会遇到很多诸如caffe,tensorflow卷积核格式、fc权重格式、输入输出数据表示格式的不同的问题,caffe-tensorflow这个项目都会自行转换。

caffe-tensorflow实现细节

caffe-tensorflow的具体原理是通过pycaffe读取网络定义文件,并重新生成一张图,图上的节点保存各个层的信息,GraphBuilder类负责生成图结构,每个节点代表一个layer,作者定义了许多transformer,每个transformer对图

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值