我遇到了一个问题,我有一个稀疏矩阵存储版本的神经网络,这意味着我只有一些参数和它的索引,比如它属于哪个层以及它属于哪个位置?我如何使用这些信息和 aprototxt
来重建caffemodel
python.Thanks
问问题
1218 次
1 回答
1
caffemodel 二进制文件只是一个二进制 proto-buffer 文件,您可以使用 proto python 接口模块 (caffe.proto.caffe_pb2) 创建/编辑它。您应该使用 caffe.io 的辅助方法将 numpy 数组转换为 protobuffer 接口(或从 protobuffer 到 numpy)可接受的数组格式。或者,您可以使用包含大多数这些方法的 caffe.Net python 类,以便更轻松地访问——您可以从 Caffe 的主页查看此示例以获取更详细的教程。
为简单起见,我将使用 prototxt 训练一个虚拟网络以创建基线 caffemodel 结构,然后使用上述方法之一使用您自己的外部权重编辑层的权重。
于 2016-03-22T10:49:07.760 回答