提交 b077e3c5 编写于 作者: L LDOUBLEV

rename

上级 af4aa1aa
......@@ -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
......
......@@ -28,7 +28,7 @@ Architecture:
model_name: large
disable_se: True
Neck:
name: CAFPN
name: RSEFPN
out_channels: 96
shortcut: True
Head:
......
......@@ -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'
]
......
......@@ -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()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册