Update predict_det.py
This commit is contained in:
parent
4c52e3857d
commit
f8441cffa4
|
@ -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):
|
||||||
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",
|
||||||
|
|
Loading…
Reference in New Issue