How to modify a .prototxt on Python? (Caffe)  ~.prototxtをPythonで編集する~

Caffeでディープラーニング研究開発をするにあたり、FeatureMapsの数だったりを増やしたり減らしたりしながら試行錯誤することはよくあります。

そんなとき、ネットワーク構造をPythonから自動的に変更させたくなりますよね?(あれ?なりませんか?)

とういわけでPythonからネットワーク構造定義ファイル.prototxtを編集する方法を調べました。こちらの投稿が参考になりました。




結論
・caffe_pb2 と google.protobuf.text_format を利用
・caffe.netから変更と保存できればと思ったけど無理そう

例えばprint(conv)とすると下記のように構造をとれていることがわかります。

 

なお、.caffemodelは

とった具合に編集していってください。

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です