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
10af9f7d
Unverified
Commit
10af9f7d
authored
Mar 02, 2021
by
Double_V
Committed by
GitHub
Mar 02, 2021
Browse files
Merge branch 'dygraph' into fix_doc
parents
c978085e
f1cd0c35
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
5 deletions
+2
-5
tools/infer/predict_det.py
tools/infer/predict_det.py
+1
-4
tools/infer_det.py
tools/infer_det.py
+1
-1
No files found.
tools/infer/predict_det.py
View file @
10af9f7d
...
@@ -39,10 +39,7 @@ class TextDetector(object):
...
@@ -39,10 +39,7 @@ class TextDetector(object):
self
.
args
=
args
self
.
args
=
args
self
.
det_algorithm
=
args
.
det_algorithm
self
.
det_algorithm
=
args
.
det_algorithm
pre_process_list
=
[{
pre_process_list
=
[{
'DetResizeForTest'
:
{
'DetResizeForTest'
:
None
'limit_side_len'
:
args
.
det_limit_side_len
,
'limit_type'
:
args
.
det_limit_type
}
},
{
},
{
'NormalizeImage'
:
{
'NormalizeImage'
:
{
'std'
:
[
0.229
,
0.224
,
0.225
],
'std'
:
[
0.229
,
0.224
,
0.225
],
...
...
tools/infer_det.py
View file @
10af9f7d
...
@@ -97,7 +97,7 @@ def main():
...
@@ -97,7 +97,7 @@ def main():
preds
=
model
(
images
)
preds
=
model
(
images
)
post_result
=
post_process_class
(
preds
,
shape_list
)
post_result
=
post_process_class
(
preds
,
shape_list
)
boxes
=
post_result
[
0
][
'points'
]
boxes
=
post_result
[
0
][
'points'
]
# write resul
e
# write resul
t
dt_boxes_json
=
[]
dt_boxes_json
=
[]
for
box
in
boxes
:
for
box
in
boxes
:
tmp_json
=
{
"transcription"
:
""
}
tmp_json
=
{
"transcription"
:
""
}
...
...
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