diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index 62e7ff73964e3c387a19e6f6a1c212ccfdbf8c91..5c75e0c480eac6796d6d4b7075d1b38d254380fd 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -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' ],