2018.12.05——caffe网络结构保存格式,如何得到TensorFlow下的.npy文件、将caffe模型转换为TensorFlow(的npy文件)模型

参数数据文件xxx.caffemodel和网络结构文件deploy.prototxt

是我等弱渣见识少了,偶然在查资料的过程中发现的,不禁惊呼竟然还有这种操作,以res-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 

执行命令之前是这样的:

2018.12.05——caffe网络结构保存格式,如何得到TensorFlow下的.npy文件、将caffe模型转换为TensorFlow(的npy文件)模型

执行命令之后是这样的:

2018.12.05——caffe网络结构保存格式,如何得到TensorFlow下的.npy文件、将caffe模型转换为TensorFlow(的npy文件)模型

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

嗯,差不多就是这样了。

原文:https://blog.****.net/sinat_35308776/article/details/79682159