提交 49c4a3a7 编写于 作者: L LDOUBLEV

gpu_id is None when args.use_gpu as False

上级 5bb6ae05
...@@ -111,7 +111,7 @@ class TextDetector(object): ...@@ -111,7 +111,7 @@ class TextDetector(object):
inference_config=self.config, inference_config=self.config,
pids=pid, pids=pid,
process_name=None, process_name=None,
gpu_ids=gpu_id, gpu_ids=gpu_id if args.use_gpu else None,
time_keys=[ time_keys=[
'preprocess_time', 'inference_time', 'postprocess_time' 'preprocess_time', 'inference_time', 'postprocess_time'
], ],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册