Tensorflow模型转换caffemodel要点

版权声明:未经本人许可,不得用于商业用途及传统媒体。转载请注明出处! https://blog.csdn.net/qikaihuting/article/details/83315795

TF model convert to caffe model


caffe 模型的定义文件

  • 即.prototxt文件,该模型的结构与TF中的要保持一致;TF的模型定义通常是inference.py文件

确定两种框架下模型中变量名与层名之间的对应关系(映射关系)

  • 可以分别写在两个txt文件中

生成caffemodel文件

  • 根据TF的模型参数文件(checkpoint),caffe的模型定义文件以及TF和caffe变量之间的映射关系,来生成.caffemodel模型参数

验证模型转换的准确性

  • 这一步非常有必要,TF和caffe的卷积和池化操作在针对某些特征图尺寸取整或者填充机制上有所不同;例如当padding=0时,TF的pooling为向下取整,caffe的pooling为向上取整。padding不为0,填充方式也不一样。

参考项目

猜你喜欢

转载自blog.csdn.net/qikaihuting/article/details/83315795