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
729ff65f
Commit
729ff65f
authored
Jun 28, 2021
by
LDOUBLEV
Browse files
Merge branch 'autolog' of
https://github.com/LDOUBLEV/PaddleOCR
into autolog
parents
b37c597d
f8441cff
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
17 additions
and
16 deletions
+17
-16
tools/infer/predict_det.py
tools/infer/predict_det.py
+17
-16
No files found.
tools/infer/predict_det.py
View file @
729ff65f
...
...
@@ -32,7 +32,6 @@ from ppocr.data import create_operators, transform
from
ppocr.postprocess
import
build_post_process
logger
=
get_logger
()
import
auto_log
class
TextDetector
(
object
):
...
...
@@ -99,10 +98,12 @@ class TextDetector(object):
self
.
predictor
,
self
.
input_tensor
,
self
.
output_tensors
,
self
.
config
=
utility
.
create_predictor
(
args
,
'det'
,
logger
)
if
args
.
benchmark
:
import
auto_log
pid
=
os
.
getpid
()
self
.
autolog
=
auto_log
.
AutoLogger
(
model_name
=
"det"
,
model_precision
=
"fp32"
,
model_precision
=
args
.
precision
,
batch_size
=
1
,
data_shape
=
"dynamic"
,
save_path
=
"./output/auto_log.lpg"
,
...
...
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