rename
parent
af4aa1aa2a
commit
b077e3c559
|
@ -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()
|
||||
|
||||
|
|
Loading…
Reference in New Issue