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

Question

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

重建 caffemodel

Answer 1

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

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