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