我想在android中运行训练有素的caffe模型。我打算使用caffe的AICamera示例,然后对其进行修改以运行我的模型。我能够编译并构建项目。
目前,我的caffe模型定义为prototxt
文件,而预训练模型为.caffemodel
文件。但是AICamera使用squeeze_init_net.pb
文件和squeeze_predict_net.pb
文件读取模型。那么如何将我必须的文件转换为.pb文件?
答案 0 :(得分:0)
您链接的AICamera示例适用于Caffe2,它与Caffe向后不兼容。
在Caffe2 Model Zoo页上,他们谈论了它:
兼容性: Caffe2采用了较新的格式,通常在 protobuf .pb文件格式,因此原始.caffemodel文件将需要 转换。
在同一页面上,他们链接到Migration page,在此处说明如何将较旧的.caffemodel转换为.pb文件格式。
基本上,他们提供了python script来将您的旧格式转换为较新的格式。另外,还有一个test script。
如果要运行.caffemodel,有两种方法(据我所知):