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
f3cf94a2
Unverified
Commit
f3cf94a2
authored
May 14, 2020
by
Double_V
Committed by
GitHub
May 14, 2020
Browse files
Merge pull request #18 from LDOUBLEV/fixocr
change the saved path of visualized image
parents
8e1f9f16
010533bd
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
1 deletion
+3
-1
tools/infer/predict_system.py
tools/infer/predict_system.py
+3
-1
No files found.
tools/infer/predict_system.py
View file @
f3cf94a2
...
...
@@ -132,9 +132,11 @@ if __name__ == "__main__":
draw_img
=
draw_ocr
(
image
,
boxes
,
txts
,
scores
,
draw_txt
=
True
,
drop_score
=
0.5
)
draw_img_save
=
"./
doc/imgs
_results/"
draw_img_save
=
"./
inference
_results/"
if
not
os
.
path
.
exists
(
draw_img_save
):
os
.
makedirs
(
draw_img_save
)
cv2
.
imwrite
(
os
.
path
.
join
(
draw_img_save
,
os
.
path
.
basename
(
image_file
)),
draw_img
)
print
(
"The visualized image saved in {}"
.
format
(
os
.
path
.
join
(
draw_img_save
,
os
.
path
.
basename
(
image_file
))))
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