从caffe训练模型生成PB文件

时间:2018-06-30 09:48:32

标签: caffe

我想在android中运行训练有素的caffe模型。我打算使用caffe的AICamera示例,然后对其进行修改以运行我的模型。我能够编译并构建项目。

目前,我的caffe模型定义为prototxt文件,而预训练模型为.caffemodel文件。但是AICamera使用squeeze_init_net.pb文件和squeeze_predict_net.pb文件读取模型。那么如何将我必须的文件转换为.pb文件?

1 个答案:

答案 0 :(得分:0)

您链接的AICamera示例适用于Caffe2,它与Caffe向后不兼容。

Caffe2 Model Zoo页上,他们谈论了它:

  

兼容性:   Caffe2采用了较新的格式,通常在   protobuf .pb文件格式,因此原始.caffemodel文件将需要   转换。

在同一页面上,他们链接到Migration page,在此处说明如何将较旧的.caffemodel转换为.pb文件格式。

基本上,他们提供了python script来将您的旧格式转换为较新的格式。另外,还有一个test script

如果要运行.caffemodel,有两种方法(据我所知):

  1. 将其转换为更新的.pb文件格式
  2. 借助OpenCV
  3. 加载.caffemodel