未验证 提交 00e961c8 编写于 作者: F Feng Ni 提交者: GitHub

remove hardnet-39 (#6487)

* remove hardnet39

* remove hardnet39
上级 49d0000c
...@@ -146,7 +146,7 @@ class HarDBlock(nn.Layer): ...@@ -146,7 +146,7 @@ class HarDBlock(nn.Layer):
class HarDNet(nn.Layer): class HarDNet(nn.Layer):
def __init__(self, depth_wise=False, return_idx=[1, 3, 8, 13], arch=85): def __init__(self, depth_wise=False, return_idx=[1, 3, 8, 13], arch=85):
super(HarDNet, self).__init__() super(HarDNet, self).__init__()
assert arch in [39, 68, 85], "HarDNet-{} not support.".format(arch) assert arch in [68, 85], "HarDNet-{} is not supported.".format(arch)
if arch == 85: if arch == 85:
first_ch = [48, 96] first_ch = [48, 96]
second_kernel = 3 second_kernel = 3
...@@ -161,6 +161,8 @@ class HarDNet(nn.Layer): ...@@ -161,6 +161,8 @@ class HarDNet(nn.Layer):
grmul = 1.7 grmul = 1.7
gr = [14, 16, 20, 40] gr = [14, 16, 20, 40]
n_layers = [8, 16, 16, 16] n_layers = [8, 16, 16, 16]
else:
raise ValueError("HarDNet-{} is not supported.".format(arch))
self.return_idx = return_idx self.return_idx = return_idx
self._out_channels = [96, 214, 458, 784] self._out_channels = [96, 214, 458, 784]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册