提交 62a01f7e 编写于 作者: L LDOUBLEV

fix

上级 d222ff03
......@@ -32,7 +32,7 @@ NetWorks:
model_name: large
disable_se: true
Neck:
name: CAPAN
name: CAFPN
out_channels: 96
shortcut: True
Head:
......@@ -48,7 +48,7 @@ NetWorks:
model_name: large
disable_se: true
Neck:
name: CAPAN
name: CAFPN
out_channels: 96
shortcut: True
Head:
......
......@@ -28,7 +28,7 @@ Architecture:
model_name: large
disable_se: True
Neck:
name: CAPAN
name: CAFPN
out_channels: 96
shortcut: True
Head:
......
......@@ -37,8 +37,8 @@ class Head(nn.Layer):
self.conv1 = nn.Conv2D(
in_channels=in_channels,
out_channels=in_channels // 4,
kernel_size=kernel_size[0],
padding=int(kernel_size[0] // 2),
kernel_size=kernel_list[0],
padding=int(kernel_list[0] // 2),
weight_attr=ParamAttr(),
bias_attr=False)
self.conv_bn1 = nn.BatchNorm(
......@@ -51,7 +51,7 @@ class Head(nn.Layer):
self.conv2 = nn.Conv2DTranspose(
in_channels=in_channels // 4,
out_channels=in_channels // 4,
kernel_size=kernel_size[1],
kernel_size=kernel_list[1],
stride=2,
weight_attr=ParamAttr(
initializer=paddle.nn.initializer.KaimingUniform()),
......@@ -66,7 +66,7 @@ class Head(nn.Layer):
self.conv3 = nn.Conv2DTranspose(
in_channels=in_channels // 4,
out_channels=1,
kernel_size=kernel_size[2],
kernel_size=kernel_list[2],
stride=2,
weight_attr=ParamAttr(
initializer=paddle.nn.initializer.KaimingUniform()),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册