如何使用python重塑caffe中的圖層？

Question

可以在prototxt文件中使用"Reshape"層。
但是，嘗試在python中使用它（使用NetSpec() ）：

n.resh = L.Reshape(n.fc3, reshape_param={'shape':'{dim:1 dim:1 dim:64 dim:64}'})

除了錯誤我什么都沒有：

 AttributeError: 'BlobShape' object has no attribute 'append' 

Answer 1

嘗試：

n.resh = L.Reshape(n.fc3, reshape_param={'shape':{'dim': [1, 1, 64, 64]}})

請注意，形狀矢量[1, 1, 64, 64] 1、1、64、64 [1, 1, 64, 64]作為列表而不是像prototxt語法那樣作為字符串傳遞。

實際上，在使用NetSpec進行接口時，應將在caffe.proto定義為repeated任何條目視為列表/向量。