从 Caffe 训练模型生成 PB 文件

  • 本文关键字:PB 文件 模型 Caffe caffe
  • 更新时间 :
  • 英文 :


我想在Android中运行一个经过训练的caffe模型。我计划通过 caffe 使用 AICamera 示例,然后对其进行修改以运行我的模型。我能够编译和构建项目。

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

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

在Caffe2 Model Zoo页面中,他们谈论它:

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

在同一页面中,他们链接到迁移页面,在那里他们解释了如何将较旧的 .caffemodel 转换为 .pb 文件格式。

基本上,他们提供了一个python脚本来将您的旧格式转换为新格式。此外,还有一个测试脚本。

如果你想运行你的.caffemodel,有两种方法(我知道(:

  1. 将其转换为较新的 .pb 文件格式
  2. 在OpenCV的帮助下加载你的.caffemodel

最新更新