将caffe模型转换为TensorFlow(的npy文件)模型

3 篇文章 0 订阅
3 篇文章 0 订阅

是我等弱渣见识少了,偶然在查资料的过程中发现的,不禁惊呼竟然还有这种操作,以resnet-50为例子,将自己的转换过程记录一下。首先附上转换过程中所要使用的程序,该程序是在GitHub上开源的,

地址为:https://github.com/ethereon/caffe-tensorflow

该项目的readme页面上还列出了使用该项目应该注意的一些事项,例如caffe的版本问题。

我们需要准备你想要转换的模型的参数数据文件xxx.caffemodel和网络结构文件deploy.prototxt。对应在我这里就是:Resnet50-model.caffemodel和Resnet_50_deploy.prototxt,我们要运行的文件是:convert.py ,根据该程序的具体要求,它的后面还要跟上几个参数,具体命令书写如下:

./convert.py Resnet_50_deploy.prototxt --caffemodel Resnet50-model.caffemodel --code-output-path=Resnet-50.py --data-output-path=Resnet-50.npy 

执行命令之前是这样的:

执行命令之后是这样的:

然后我们就得到了TensorFlow下的.npy文件。

嗯,差不多就是这样了。

 

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值