提交 85dbe94c 编写于 作者: S SunAhong1993

fix the prelu

上级 108533b2
...@@ -243,7 +243,10 @@ class CaffeDecoder(object): ...@@ -243,7 +243,10 @@ class CaffeDecoder(object):
for blob in layer.blobs: for blob in layer.blobs:
if len(blob.shape.dim): if len(blob.shape.dim):
dims = 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: else:
c_o = blob.num c_o = blob.num
c_i = blob.channels c_i = blob.channels
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册