Skip to content

GitLab

Unverified Commit f8441cff authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Update predict_det.py

parent 4c52e385
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/infer/predict_det.py
View file @ f8441cff
...@@ -34,7 +34,6 @@ from ppocr.postprocess import build_post_process ...@@ -34,7 +34,6 @@ from ppocr.postprocess import build_post_process
# import tools.infer.benchmark_utils as benchmark_utils # import tools.infer.benchmark_utils as benchmark_utils
logger = get_logger() logger = get_logger()
import auto_log
class TextDetector(object): class TextDetector(object):
...@@ -102,6 +101,7 @@ class TextDetector(object): ...@@ -102,6 +101,7 @@ class TextDetector(object):
args, 'det', logger) args, 'det', logger)
if args.benchmark: if args.benchmark:
import auto_log
pid = os.getpid() pid = os.getpid()
self.autolog = auto_log.AutoLogger( self.autolog = auto_log.AutoLogger(
model_name="det", model_name="det",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment