未验证 提交 d59c80c1 编写于 作者: J Jason 提交者: GitHub

Merge pull request #192 from SunAhong1993/develop

fix the PReLU
...@@ -243,7 +243,12 @@ class CaffeDecoder(object): ...@@ -243,7 +243,12 @@ 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.
先完成此消息的编辑!
想要评论请 注册