新しい配列を.caffemodelファイルにダンプする

Question

カーネルサイズ、パッド情報などの特定のパラメータを持つカスタマイズされたレイヤーがあります。この中で、bool/int型の配列パラメータがあります。処理後にこの配列パラメータを.caffemodelファイルに追加する必要があります。アレイをカフェモデルファイルにリンク/ダンプする方法は？

以下のように、配列パラメータを.protoファイルに含めました。

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

bool型のblobを作成することはできますか？

Answer 1

あなたはPY-カフェ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。