如何使用Python從caffe中的.prototxt中刪除圖層

Question

我使用Python代碼創建了train.prototxt，並且想要刪除loss圖層以自動創建deploy.prototxt。但是，我只知道用這樣的整數去除圖層的方法：

net_param = deploy_net.to_proto() 
del net_param.layer[0] 

有沒有可能通過名稱刪除圖層？ Python API的文檔在哪裏？我找不到它。我只需要查看C++代碼並嘗試將其轉換爲Python代碼？

編輯

我初始化與網。

net = caffe.NetSpec() 

Answer 1

net.layer_dict是所有的層的一個字典。因此，要刪除，你可以這樣做：

del net.layer_dict['layer_name'];

你可以看看pycaffe.py對於Python API中的細節。