gpu_id is None when args.use_gpu as False
This commit is contained in:
parent
5bb6ae05c4
commit
49c4a3a7ed
|
@ -111,7 +111,7 @@ class TextDetector(object):
|
|||
inference_config=self.config,
|
||||
pids=pid,
|
||||
process_name=None,
|
||||
gpu_ids=gpu_id,
|
||||
gpu_ids=gpu_id if args.use_gpu else None,
|
||||
time_keys=[
|
||||
'preprocess_time', 'inference_time', 'postprocess_time'
|
||||
],
|
||||
|
|
Loading…
Reference in New Issue