Skip to content

GitLab

Unverified Commit 4c52e385 authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Update predict_det.py

parent ea60a034
Show whitespace changes
Inline Side-by-side
Showing with 16 additions and 15 deletions
tools/infer/predict_det.py
View file @ 4c52e385
......@@ -101,10 +101,11 @@ class TextDetector(object):
self.predictor, self.input_tensor, self.output_tensors, self.config = utility.create_predictor(
args, 'det', logger)
if args.benchmark:
pid = os.getpid()
self.autolog = auto_log.AutoLogger(
model_name="det",
model_precision="fp32",
model_precision=args.precision,
batch_size=1,
data_shape="dynamic",
save_path="./output/auto_log.lpg",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment