提交 729ff65f 编写于 作者: L LDOUBLEV

Merge branch 'autolog' of https://github.com/LDOUBLEV/PaddleOCR into autolog

...@@ -32,7 +32,6 @@ from ppocr.data import create_operators, transform ...@@ -32,7 +32,6 @@ from ppocr.data import create_operators, transform
from ppocr.postprocess import build_post_process from ppocr.postprocess import build_post_process
logger = get_logger() logger = get_logger()
import auto_log
class TextDetector(object): class TextDetector(object):
...@@ -99,21 +98,23 @@ class TextDetector(object): ...@@ -99,21 +98,23 @@ class TextDetector(object):
self.predictor, self.input_tensor, self.output_tensors, self.config = utility.create_predictor( self.predictor, self.input_tensor, self.output_tensors, self.config = utility.create_predictor(
args, 'det', logger) args, 'det', logger)
pid = os.getpid() if args.benchmark:
self.autolog = auto_log.AutoLogger( import auto_log
model_name="det", pid = os.getpid()
model_precision="fp32", self.autolog = auto_log.AutoLogger(
batch_size=1, model_name="det",
data_shape="dynamic", model_precision=args.precision,
save_path="./output/auto_log.lpg", batch_size=1,
inference_config=self.config, data_shape="dynamic",
pids=pid, save_path="./output/auto_log.lpg",
process_name=None, inference_config=self.config,
gpu_ids=0, pids=pid,
time_keys=[ process_name=None,
'preprocess_time', 'inference_time', 'postprocess_time' gpu_ids=0,
], time_keys=[
warmup=10) 'preprocess_time', 'inference_time', 'postprocess_time'
],
warmup=10)
def order_points_clockwise(self, pts): def order_points_clockwise(self, pts):
""" """
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册