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
a35976f0
"vscode:/vscode.git/clone" did not exist on "2e9abcb91d058af87b188aee5e7efd1cf34eead0"
Unverified
Commit
a35976f0
authored
Oct 14, 2021
by
Double_V
Committed by
GitHub
Oct 14, 2021
Browse files
Merge pull request #4308 from LDOUBLEV/dygraph
fix BGR
parents
89c9f363
008876d4
Changes
6
Hide whitespace changes
Inline
Side-by-side
Showing
6 changed files
with
7 additions
and
6 deletions
+7
-6
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/modeling/necks/rnn.py
ppocr/modeling/necks/rnn.py
+1
-1
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 @
a35976f0
...
...
@@ -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 @
a35976f0
...
...
@@ -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 @
a35976f0
...
...
@@ -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 @
a35976f0
...
...
@@ -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/modeling/necks/rnn.py
View file @
a35976f0
...
...
@@ -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
,
...
...
tools/infer/predict_e2e.py
View file @
a35976f0
...
...
@@ -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