Update predict_det.py

This commit is contained in:
Double_V 2021-06-24 17:48:58 +08:00 committed by GitHub
parent 4c52e3857d
commit f8441cffa4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -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",