pull/6043/head
LDOUBLEV 2022-04-27 19:57:48 +08:00
parent af4aa1aa2a
commit b077e3c559
4 changed files with 11 additions and 11 deletions

View File

@ -33,7 +33,7 @@ Architecture:
model_name: large
disable_se: true
Neck:
name: CAFPN
name: RSEFPN
out_channels: 96
shortcut: True
Head:
@ -49,7 +49,7 @@ Architecture:
model_name: large
disable_se: true
Neck:
name: CAFPN
name: RSEFPN
out_channels: 96
shortcut: True
Head:
@ -65,7 +65,7 @@ Architecture:
in_channels: 3
layers: 50
Neck:
name: FEPAN
name: LKPAN
out_channels: 256
Head:
name: DBHead

View File

@ -28,7 +28,7 @@ Architecture:
model_name: large
disable_se: True
Neck:
name: CAFPN
name: RSEFPN
out_channels: 96
shortcut: True
Head:

View File

@ -26,7 +26,7 @@ def build_neck(config):
from .fce_fpn import FCEFPN
from .pren_fpn import PRENFPN
support_dict = [
'FPN', 'FCEFPN', 'FEPAN', 'DBFPN', 'CAFPN', 'EASTFPN', 'SASTFPN',
'FPN', 'FCEFPN', 'LKPAN', 'DBFPN', 'RSEFPN', 'EASTFPN', 'SASTFPN',
'SequenceEncoder', 'PGFPN', 'TableFPN', 'PRENFPN'
]

View File

@ -190,9 +190,9 @@ class DBFPN(nn.Layer):
return fuse
class CALayer(nn.Layer):
class RSELayer(nn.Layer):
def __init__(self, in_channels, out_channels, kernel_size, shortcut=True):
super(CALayer, self).__init__()
super(RSELayer, self).__init__()
weight_attr = paddle.nn.initializer.KaimingUniform()
self.out_channels = out_channels
self.in_conv = nn.Conv2D(
@ -214,9 +214,9 @@ class CALayer(nn.Layer):
return out
class CAFPN(nn.Layer):
class RSEFPN(nn.Layer):
def __init__(self, in_channels, out_channels, shortcut=True, **kwargs):
super(CAFPN, self).__init__()
super(RSEFPN, self).__init__()
self.out_channels = out_channels
self.ins_conv = nn.LayerList()
self.inp_conv = nn.LayerList()
@ -263,9 +263,9 @@ class CAFPN(nn.Layer):
return fuse
class FEPAN(nn.Layer):
class LKPAN(nn.Layer):
def __init__(self, in_channels, out_channels, mode='large', **kwargs):
super(FEPAN, self).__init__()
super(LKPAN, self).__init__()
self.out_channels = out_channels
weight_attr = paddle.nn.initializer.KaimingUniform()