我想在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,有两种方法(我知道(:
- 将其转换为较新的 .pb 文件格式
- 在OpenCV的帮助下加载你的.caffemodel 。