forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update predict_det.py

This commit is contained in:
Double_V 2021-06-24 17:48:36 +08:00 committed by GitHub
parent ea60a03487
commit 4c52e3857d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
tools/infer/predict_det.py
View File

@ -101,21 +101,22 @@ 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( pid = os.getpid()
model_name="det", self.autolog = auto_log.AutoLogger(
model_precision="fp32", model_name="det",
batch_size=1, model_precision=args.precision,
data_shape="dynamic", batch_size=1,
save_path="./output/auto_log.lpg", data_shape="dynamic",
inference_config=self.config, save_path="./output/auto_log.lpg",
pids=pid, inference_config=self.config,
process_name=None, pids=pid,
gpu_ids=0, process_name=None,
time_keys=[ gpu_ids=0,
'preprocess_time', 'inference_time', 'postprocess_time' time_keys=[
], 'preprocess_time', 'inference_time', 'postprocess_time'
warmup=10) ],
warmup=10)
def order_points_clockwise(self, pts): def order_points_clockwise(self, pts):
""" """