打个广告 欢迎大家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 LayerAdapter(object)

本文手把手教你如何将DenseNet_161的Caffe模型转换为TensorFlow模型。首先确保使用Python 2.7和TensorFlow 1.0以上版本。然后下载必要的文件,包括Caffe模型和转换脚本。修改prototxt文件,删除特定行。接着修改转换脚本中的几个关键文件,包括 layers.py、shapes.py 和 transformers.py。完成这些步骤后,将生成TensorFlow的网络结构py文件和参数npy文件。

1803

被折叠的 条评论
为什么被折叠?



