Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
b077e3c5
"...git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "cf1385a7f29ea52b510f31f3214cd8f8500c6afd"
Commit
b077e3c5
authored
Apr 27, 2022
by
LDOUBLEV
Browse files
rename
parent
af4aa1aa
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
11 additions
and
11 deletions
+11
-11
configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml
configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml
+3
-3
configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml
configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml
+1
-1
ppocr/modeling/necks/__init__.py
ppocr/modeling/necks/__init__.py
+1
-1
ppocr/modeling/necks/db_fpn.py
ppocr/modeling/necks/db_fpn.py
+6
-6
No files found.
configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_cml.yml
View file @
b077e3c5
...
@@ -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
:
CA
FPN
name
:
RSE
FPN
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
:
CA
FPN
name
:
RSE
FPN
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
:
FE
PAN
name
:
LK
PAN
out_channels
:
256
out_channels
:
256
Head
:
Head
:
name
:
DBHead
name
:
DBHead
...
...
configs/det/ch_PP-OCRv3/ch_PP-OCRv3_det_student.yml
View file @
b077e3c5
...
@@ -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
:
CA
FPN
name
:
RSE
FPN
out_channels
:
96
out_channels
:
96
shortcut
:
True
shortcut
:
True
Head
:
Head
:
...
...
ppocr/modeling/necks/__init__.py
View file @
b077e3c5
...
@@ -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'
,
'
FE
PAN'
,
'DBFPN'
,
'
CA
FPN'
,
'EASTFPN'
,
'SASTFPN'
,
'FPN'
,
'FCEFPN'
,
'
LK
PAN'
,
'DBFPN'
,
'
RSE
FPN'
,
'EASTFPN'
,
'SASTFPN'
,
'SequenceEncoder'
,
'PGFPN'
,
'TableFPN'
,
'PRENFPN'
'SequenceEncoder'
,
'PGFPN'
,
'TableFPN'
,
'PRENFPN'
]
]
...
...
ppocr/modeling/necks/db_fpn.py
View file @
b077e3c5
...
@@ -190,9 +190,9 @@ class DBFPN(nn.Layer):
...
@@ -190,9 +190,9 @@ class DBFPN(nn.Layer):
return
fuse
return
fuse
class
CA
Layer
(
nn
.
Layer
):
class
RSE
Layer
(
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
(
CA
Layer
,
self
).
__init__
()
super
(
RSE
Layer
,
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
CA
FPN
(
nn
.
Layer
):
class
RSE
FPN
(
nn
.
Layer
):
def
__init__
(
self
,
in_channels
,
out_channels
,
shortcut
=
True
,
**
kwargs
):
def
__init__
(
self
,
in_channels
,
out_channels
,
shortcut
=
True
,
**
kwargs
):
super
(
CA
FPN
,
self
).
__init__
()
super
(
RSE
FPN
,
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
FE
PAN
(
nn
.
Layer
):
class
LK
PAN
(
nn
.
Layer
):
def
__init__
(
self
,
in_channels
,
out_channels
,
mode
=
'large'
,
**
kwargs
):
def
__init__
(
self
,
in_channels
,
out_channels
,
mode
=
'large'
,
**
kwargs
):
super
(
FE
PAN
,
self
).
__init__
()
super
(
LK
PAN
,
self
).
__init__
()
self
.
out_channels
=
out_channels
self
.
out_channels
=
out_channels
weight_attr
=
paddle
.
nn
.
initializer
.
KaimingUniform
()
weight_attr
=
paddle
.
nn
.
initializer
.
KaimingUniform
()
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment