繁体   English   中英

如何使用python重塑caffe中的图层?

[英]How to reshape layer in caffe with python?

可以在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' 

尝试:

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任何条目视为列表/向量。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM