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
59d50c91
Unverified
Commit
59d50c91
authored
Jan 25, 2022
by
ioracion
Committed by
GitHub
Jan 25, 2022
Browse files
Update predict_system.py
parent
3609d0b2
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
tools/infer/predict_system.py
tools/infer/predict_system.py
+3
-3
No files found.
tools/infer/predict_system.py
View file @
59d50c91
...
...
@@ -92,11 +92,11 @@ class TextSystem(object):
self
.
draw_crop_rec_res
(
self
.
args
.
crop_res_save_dir
,
img_crop_list
,
rec_res
)
filter_boxes
,
filter_rec_res
=
[],
[]
for
box
,
rec_re
u
slt
in
zip
(
dt_boxes
,
rec_res
):
text
,
score
=
rec_re
u
slt
for
box
,
rec_res
u
lt
in
zip
(
dt_boxes
,
rec_res
):
text
,
score
=
rec_res
u
lt
if
score
>=
self
.
drop_score
:
filter_boxes
.
append
(
box
)
filter_rec_res
.
append
(
rec_re
u
slt
)
filter_rec_res
.
append
(
rec_res
u
lt
)
return
filter_boxes
,
filter_rec_res
...
...
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