diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index 62e7ff73..5c75e0c4 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' ],