如何用python重塑caffe中的图层？

Question

可以在原型文件中使用"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]作为列表传递，而不是像原型文本语法中那样传递。

事实上，repeated中定义为caffe.proto的任何条目在使用NetSpec进行接口时都应被视为列表/向量。