如何通过一些参数重建caffemodel？

Question

我遇到了一个问题，我有一个稀疏矩阵存储版本的神经网络，这意味着我只有一些参数及其索引，就像它属于哪个层以及它属于哪个位置？我如何使用这些信息和prototxt通过python重建caffemodel。谢谢

Answer 1

caffemodel二进制文件只是一个二进制原始缓冲区文件，您可以使用proto python interfrace模块（caffe.proto.caffe_pb2）创建/编辑该文件。您应该使用caffe.io的帮助方法将numpy数组转换为protobuffer接口（或从protobuffer到numpy）可接受的数组格式。 或者，您可以使用包含大多数这些方法的caffe.Net python类以便于访问 - 您可以查看this example from Caffe's homepage以获取更详细的教程。

为简单起见，我将使用原型文本训练虚拟网络来创建基线caffemodel结构，然后使用上述方法之一使用自己的外部权重编辑图层的权重。