将新数组转储到.caffemodel文件中

Question

我有一个带有某些参数的自定义图层，如内核大小，pad信息等。其中我有数组参数，类型为bool / int。我需要在处理后在.caffemodel文件中添加此数组参数。如何将数组链接/转储到caffemodel文件？

我在.proto文件中包含了数组参数，如下所示。

{
 ... //other parameters
 repeated bool/int <varibale_name> [packed = true]
}

是否可以创建bool类型的blob？

Answer 1

你可以通过py-caffe net surgery -

来实现

net.params[learnable_layer_name].add_blob()
new_blob_index = len(net.params[learnable_layer_name]) - 1
net.params[learnable_layer_name][new_blob_index].reshape(desired_shape)
net.params[learnable_layer_name][new_blob_index].data[:] = new_data_to_insert

HTH。