c++ - Caffe C++ 保存网络caffemodel文件

Question

我已经成功构建并训练了 audioCaffe 演示，但该演示没有保存网络。

我在 MatLab 中找到了用 Python 保存网络的文档，但我找不到任何关于 C++ 的文档。

我认为会有类似 net.save("file.caffemodel") 的功能，但我试过了，但没有成功。

在 caffe.cpp 的训练函数中，我尝试了这个:

  if (FLAGS_snapshot.size()) {
    LOG(INFO) << "Resuming from " << FLAGS_snapshot;
    solver.Solve(FLAGS_snapshot);
  } else if (FLAGS_weights.size()) {
    LOG(INFO) << "Finetuning from " << FLAGS_weights;
    solver.net()->CopyTrainedLayersFrom(FLAGS_weights);
    solver.Solve();
  } else {
    solver.Solve();
  }
  solver.save("file.caffemodel")

但是我得到了一个no method exists error

有什么想法吗？

Answer 1

请试试这个...

caffe::NetParameter net_param;
net_->ToProto(&net_param);
caffe::WriteProtoToBinaryFile(net_param, caffe_model_path);