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