本文手把手教你如何将DenseNet_161的Caffe模型转换为TensorFlow模型。首先确保使用Python 2.7和TensorFlow 1.0以上版本。然后下载必要的文件，包括Caffe模型和转换脚本。修改prototxt文件，删除特定行。接着修改转换脚本中的几个关键文件，包括 layers.py、shapes.py 和 transformers.py。完成这些步骤后，将生成TensorFlow的网络结构py文件和参数npy文件。
打个广告 欢迎大家star我的github项目，准备测试结果后上传网络和参数，可以直接拿去使用。
环境及文件
python 2.7（必须用2.7版本，3.x版本会报错）
tensorflow 1.0及其以上（我是用的1.2.0）
caffemodel文件及其prototxt文件，可以在这里下载
下载caffe to tensorflow的py文件，可以在这里下载
开始转换
这里以DenseNet_161为例子
修改prototxt文件
打开DenseNet_161.prototxt文件，删除ceil_mode: false，第54行，保存修改。
修改py文件
首先解压下载的zip文件，完成后应该是这个样子
/kaffe/caffe/目录下找到caffepb.py和resolver.py文件
将caffepb.py文件重新命名为caffe_pb2.py文件，打开resolver.py文件，将下面代码
from . import caffepb
self.caffepb = caffepb改为
from . import caffe_pb2
self.caffepb = caffe_pb2保存修改退出
回到/kaffe/目录下
打开layers.py文件, 用下面代码替换原来的代码
class 
最低0.47元/天 解锁文章
扫一扫
7372
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
