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

revise transform and pascalvoc dataset

上级 1bed29c3
......@@ -104,9 +104,6 @@ class LayerWarp(nn.Layer):
return y
DarkNet_cfg = {53: ([1, 2, 8, 8, 4])}
class DarkNet53_conv_body(nn.Layer):
"""Darknet53
Args:
......@@ -115,7 +112,7 @@ class DarkNet53_conv_body(nn.Layer):
"""
def __init__(self, ch_in: int = 3, is_test: bool = False):
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.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.
先完成此消息的编辑!
想要评论请 注册