提交 1b2b70f7 编写于 作者: M mamingjie-China

Merge branch 'develop' of https://github.com/mamingjie-China/X2Paddle into develop

......@@ -243,7 +243,12 @@ class CaffeDecoder(object):
for blob in layer.blobs:
if len(blob.shape.dim):
dims = blob.shape.dim
c_o, c_i, h, w = map(int, [1] * (4 - len(dims)) + list(dims))
if layer.type == 'PReLU':
c_o, c_i, h, w = map(int, [1] + \
list(dims) + [1]* (3 - len(dims)))
else:
c_o, c_i, h, w = map(int, [1] * (4 - len(dims)) \
+ list(dims))
else:
c_o = blob.num
c_i = blob.channels
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册