1. 程式人生 > >Tensorflow模型轉換caffemodel要點

Tensorflow模型轉換caffemodel要點

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,填充方式也不一樣。

參考專案