提交 b077e3c5 编写于 作者: L LDOUBLEV

rename

上级 af4aa1aa
...@@ -33,7 +33,7 @@ Architecture: ...@@ -33,7 +33,7 @@ Architecture:
model_name: large model_name: large
disable_se: true disable_se: true
Neck: Neck:
name: CAFPN name: RSEFPN
out_channels: 96 out_channels: 96
shortcut: True shortcut: True
Head: Head:
...@@ -49,7 +49,7 @@ Architecture: ...@@ -49,7 +49,7 @@ Architecture:
model_name: large model_name: large
disable_se: true disable_se: true
Neck: Neck:
name: CAFPN name: RSEFPN
out_channels: 96 out_channels: 96
shortcut: True shortcut: True
Head: Head:
...@@ -65,7 +65,7 @@ Architecture: ...@@ -65,7 +65,7 @@ Architecture:
in_channels: 3 in_channels: 3
layers: 50 layers: 50
Neck: Neck:
name: FEPAN name: LKPAN
out_channels: 256 out_channels: 256
Head: Head:
name: DBHead name: DBHead
......
...@@ -28,7 +28,7 @@ Architecture: ...@@ -28,7 +28,7 @@ Architecture:
model_name: large model_name: large
disable_se: True disable_se: True
Neck: Neck:
name: CAFPN name: RSEFPN
out_channels: 96 out_channels: 96
shortcut: True shortcut: True
Head: Head:
......
...@@ -26,7 +26,7 @@ def build_neck(config): ...@@ -26,7 +26,7 @@ def build_neck(config):
from .fce_fpn import FCEFPN from .fce_fpn import FCEFPN
from .pren_fpn import PRENFPN from .pren_fpn import PRENFPN
support_dict = [ support_dict = [
'FPN', 'FCEFPN', 'FEPAN', 'DBFPN', 'CAFPN', 'EASTFPN', 'SASTFPN', 'FPN', 'FCEFPN', 'LKPAN', 'DBFPN', 'RSEFPN', 'EASTFPN', 'SASTFPN',
'SequenceEncoder', 'PGFPN', 'TableFPN', 'PRENFPN' 'SequenceEncoder', 'PGFPN', 'TableFPN', 'PRENFPN'
] ]
......
...@@ -190,9 +190,9 @@ class DBFPN(nn.Layer): ...@@ -190,9 +190,9 @@ class DBFPN(nn.Layer):
return fuse return fuse
class CALayer(nn.Layer): class RSELayer(nn.Layer):
def __init__(self, in_channels, out_channels, kernel_size, shortcut=True): 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() weight_attr = paddle.nn.initializer.KaimingUniform()
self.out_channels = out_channels self.out_channels = out_channels
self.in_conv = nn.Conv2D( self.in_conv = nn.Conv2D(
...@@ -214,9 +214,9 @@ class CALayer(nn.Layer): ...@@ -214,9 +214,9 @@ class CALayer(nn.Layer):
return out return out
class CAFPN(nn.Layer): class RSEFPN(nn.Layer):
def __init__(self, in_channels, out_channels, shortcut=True, **kwargs): def __init__(self, in_channels, out_channels, shortcut=True, **kwargs):
super(CAFPN, self).__init__() super(RSEFPN, self).__init__()
self.out_channels = out_channels self.out_channels = out_channels
self.ins_conv = nn.LayerList() self.ins_conv = nn.LayerList()
self.inp_conv = nn.LayerList() self.inp_conv = nn.LayerList()
...@@ -263,9 +263,9 @@ class CAFPN(nn.Layer): ...@@ -263,9 +263,9 @@ class CAFPN(nn.Layer):
return fuse return fuse
class FEPAN(nn.Layer): class LKPAN(nn.Layer):
def __init__(self, in_channels, out_channels, mode='large', **kwargs): def __init__(self, in_channels, out_channels, mode='large', **kwargs):
super(FEPAN, self).__init__() super(LKPAN, self).__init__()
self.out_channels = out_channels self.out_channels = out_channels
weight_attr = paddle.nn.initializer.KaimingUniform() weight_attr = paddle.nn.initializer.KaimingUniform()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册