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
|
||||
|
||||
logger = get_logger()
|
||||
import auto_log
|
||||
|
||||
|
||||
class TextDetector(object):
|
||||
|
@ -102,6 +101,7 @@ class TextDetector(object):
|
|||
args, 'det', logger)
|
||||
|
||||
if args.benchmark:
|
||||
import auto_log
|
||||
pid = os.getpid()
|
||||
self.autolog = auto_log.AutoLogger(
|
||||
model_name="det",
|
||||
|
|
Loading…
Reference in New Issue