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
2400bacf
Unverified
Commit
2400bacf
authored
Oct 14, 2021
by
Bin Lu
Committed by
GitHub
Oct 14, 2021
Browse files
Merge branch 'PaddlePaddle:dygraph' into dygraph
parents
bb253d6d
37eec4d5
Changes
9
Hide whitespace changes
Inline
Side-by-side
Showing
9 changed files
with
109 additions
and
15 deletions
+109
-15
configs/det/ch_PP-OCRv2/ch_PP-OCR_det_cml.yml
configs/det/ch_PP-OCRv2/ch_PP-OCR_det_cml.yml
+1
-0
configs/det/ch_PP-OCRv2/ch_PP-OCR_det_distill.yml
configs/det/ch_PP-OCRv2/ch_PP-OCR_det_distill.yml
+2
-2
configs/det/ch_PP-OCRv2/ch_PP-OCR_det_dml.yml
configs/det/ch_PP-OCRv2/ch_PP-OCR_det_dml.yml
+1
-0
configs/e2e/e2e_r50_vd_pg.yml
configs/e2e/e2e_r50_vd_pg.yml
+2
-2
ppocr/losses/rec_sar_loss.py
ppocr/losses/rec_sar_loss.py
+6
-3
ppocr/modeling/necks/rnn.py
ppocr/modeling/necks/rnn.py
+1
-1
ppocr/utils/dict90.txt
ppocr/utils/dict90.txt
+90
-0
ppstructure/layout/train_layoutparser_model.md
ppstructure/layout/train_layoutparser_model.md
+6
-6
tools/infer/predict_e2e.py
tools/infer/predict_e2e.py
+0
-1
No files found.
configs/det/ch_PP-OCRv2/ch_PP-OCR_det_cml.yml
View file @
2400bacf
...
...
@@ -141,6 +141,7 @@ Train:
img_mode
:
BGR
channel_first
:
False
-
DetLabelEncode
:
# Class handling label
-
CopyPaste
:
-
IaaAugment
:
augmenter_args
:
-
{
'
type'
:
Fliplr
,
'
args'
:
{
'
p'
:
0.5
}
}
...
...
configs/det/ch_PP-OCRv2/ch_PP-OCR_det_distill.yml
View file @
2400bacf
...
...
@@ -68,8 +68,7 @@ Loss:
ohem_ratio
:
3
-
DistillationDBLoss
:
weight
:
1.0
model_name_list
:
[
"
Student"
,
"
Teacher"
]
# key: maps
model_name_list
:
[
"
Student"
]
name
:
DBLoss
balance_loss
:
true
main_loss_type
:
DiceLoss
...
...
@@ -116,6 +115,7 @@ Train:
img_mode
:
BGR
channel_first
:
False
-
DetLabelEncode
:
# Class handling label
-
CopyPaste
:
-
IaaAugment
:
augmenter_args
:
-
{
'
type'
:
Fliplr
,
'
args'
:
{
'
p'
:
0.5
}
}
...
...
configs/det/ch_PP-OCRv2/ch_PP-OCR_det_dml.yml
View file @
2400bacf
...
...
@@ -118,6 +118,7 @@ Train:
img_mode
:
BGR
channel_first
:
False
-
DetLabelEncode
:
# Class handling label
-
CopyPaste
:
-
IaaAugment
:
augmenter_args
:
-
{
'
type'
:
Fliplr
,
'
args'
:
{
'
p'
:
0.5
}
}
...
...
configs/e2e/e2e_r50_vd_pg.yml
View file @
2400bacf
...
...
@@ -94,7 +94,7 @@ Eval:
label_file_list
:
[
./train_data/total_text/test/test.txt
]
transforms
:
-
DecodeImage
:
# load image
img_mode
:
RGB
img_mode
:
BGR
channel_first
:
False
-
E2ELabelEncodeTest
:
-
E2EResizeForTest
:
...
...
@@ -111,4 +111,4 @@ Eval:
shuffle
:
False
drop_last
:
False
batch_size_per_card
:
1
# must be 1
num_workers
:
2
\ No newline at end of file
num_workers
:
2
ppocr/losses/rec_sar_loss.py
View file @
2400bacf
...
...
@@ -9,11 +9,14 @@ from paddle import nn
class
SARLoss
(
nn
.
Layer
):
def
__init__
(
self
,
**
kwargs
):
super
(
SARLoss
,
self
).
__init__
()
self
.
loss_func
=
paddle
.
nn
.
loss
.
CrossEntropyLoss
(
reduction
=
"mean"
,
ignore_index
=
96
)
self
.
loss_func
=
paddle
.
nn
.
loss
.
CrossEntropyLoss
(
reduction
=
"mean"
,
ignore_index
=
92
)
def
forward
(
self
,
predicts
,
batch
):
predict
=
predicts
[:,
:
-
1
,
:]
# ignore last index of outputs to be in same seq_len with targets
label
=
batch
[
1
].
astype
(
"int64"
)[:,
1
:]
# ignore first index of target in loss calculation
predict
=
predicts
[:,
:
-
1
,
:]
# ignore last index of outputs to be in same seq_len with targets
label
=
batch
[
1
].
astype
(
"int64"
)[:,
1
:]
# ignore first index of target in loss calculation
batch_size
,
num_steps
,
num_classes
=
predict
.
shape
[
0
],
predict
.
shape
[
1
],
predict
.
shape
[
2
]
assert
len
(
label
.
shape
)
==
len
(
list
(
predict
.
shape
))
-
1
,
\
...
...
ppocr/modeling/necks/rnn.py
View file @
2400bacf
...
...
@@ -51,7 +51,7 @@ class EncoderWithFC(nn.Layer):
super
(
EncoderWithFC
,
self
).
__init__
()
self
.
out_channels
=
hidden_size
weight_attr
,
bias_attr
=
get_para_bias_attr
(
l2_decay
=
0.00001
,
k
=
in_channels
,
name
=
'reduce_encoder_fea'
)
l2_decay
=
0.00001
,
k
=
in_channels
)
self
.
fc
=
nn
.
Linear
(
in_channels
,
hidden_size
,
...
...
ppocr/utils/dict90.txt
0 → 100644
View file @
2400bacf
0
1
2
3
4
5
6
7
8
9
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
A
B
C
D
E
F
G
H
I
J
K
L
M
N
O
P
Q
R
S
T
U
V
W
X
Y
Z
!
"
#
$
%
&
'
(
)
*
+
,
-
.
/
:
;
<
=
>
?
@
[
\
]
_
`
~
\ No newline at end of file
ppstructure/layout/train_layoutparser_model.md
View file @
2400bacf
...
...
@@ -4,9 +4,9 @@
[
1.1 Requirements
](
#Requirements
)
[
1.2 Install PaddleDetection
](
#Install
PaddleDetection)
[
1.2 Install PaddleDetection
](
#Install
_
PaddleDetection
)
[
2. Data preparation
](
#Data
p
reparation)
[
2. Data preparation
](
#Data
_
reparation
)
[
3. Configuration
](
#Configuration
)
...
...
@@ -16,7 +16,7 @@
[
6. Deployment
](
#Deployment
)
[
6.1 Export model
](
#Export
model)
[
6.1 Export model
](
#Export
_
model
)
[
6.2 Inference
](
#Inference
)
...
...
@@ -35,7 +35,7 @@
-
CUDA >= 10.1
-
cuDNN >= 7.6
<a
name=
"Install
PaddleDetection"
></a>
<a
name=
"Install
_
PaddleDetection"
></a>
### 1.2 Install PaddleDetection
...
...
@@ -51,7 +51,7 @@ pip install -r requirements.txt
For more installation tutorials, please refer to:
[
Install doc
](
https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.1/docs/tutorials/INSTALL_cn.md
)
<a
name=
"Data
preparation"
></a>
<a
name=
"Data
_
preparation"
></a>
## 2. Data preparation
...
...
@@ -165,7 +165,7 @@ python tools/infer.py -c configs/ppyolo/ppyolov2_r50vd_dcn_365e_coco.yml --infer
Use your trained model in Layout Parser
<a
name=
"Export
model"
></a>
<a
name=
"Export
_
model"
></a>
### 6.1 Export model
...
...
tools/infer/predict_e2e.py
View file @
2400bacf
...
...
@@ -141,7 +141,6 @@ if __name__ == "__main__":
img
,
flag
=
check_and_read_gif
(
image_file
)
if
not
flag
:
img
=
cv2
.
imread
(
image_file
)
img
=
img
[:,
:,
::
-
1
]
if
img
is
None
:
logger
.
info
(
"error in loading image:{}"
.
format
(
image_file
))
continue
...
...
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