提交 2f8ff8e6 编写于 作者: H haoyuying

revise transform and pascalvoc dataset

上级 1bed29c3
...@@ -104,9 +104,6 @@ class LayerWarp(nn.Layer): ...@@ -104,9 +104,6 @@ class LayerWarp(nn.Layer):
return y return y
DarkNet_cfg = {53: ([1, 2, 8, 8, 4])}
class DarkNet53_conv_body(nn.Layer): class DarkNet53_conv_body(nn.Layer):
"""Darknet53 """Darknet53
Args: Args:
...@@ -115,7 +112,7 @@ class DarkNet53_conv_body(nn.Layer): ...@@ -115,7 +112,7 @@ class DarkNet53_conv_body(nn.Layer):
""" """
def __init__(self, ch_in: int = 3, is_test: bool = False): def __init__(self, ch_in: int = 3, is_test: bool = False):
super(DarkNet53_conv_body, self).__init__() super(DarkNet53_conv_body, self).__init__()
self.stages = DarkNet_cfg[53] self.stages = [1, 2, 8, 8, 4]
self.stages = self.stages[0:5] self.stages = self.stages[0:5]
self.conv0 = ConvBNLayer(ch_in=ch_in, ch_out=32, filter_size=3, stride=1, padding=1, is_test=is_test) self.conv0 = ConvBNLayer(ch_in=ch_in, ch_out=32, filter_size=3, stride=1, padding=1, is_test=is_test)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册