Merge pull request #611 from t00350320/main

HumanParsing add 'CUDAExecutionProvider'
yolain · Jan 10, 2025 · be8306b · be8306b
2 parents 5e0cc2e + 5ec7449
commit be8306b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py/human_parsing/run_parsing.py b/py/human_parsing/run_parsing.py
@@ -19,7 +19,7 @@ def __call__(self, input_image, mask_components):
             session_options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL
             # session_options.add_session_config_entry('gpu_id', str(gpu_id))
             self.session = ort.InferenceSession(self.model_path, sess_options=session_options,
-                                                providers=['CPUExecutionProvider'])
+                                                providers=['CUDAExecutionProvider', 'CPUExecutionProvider'])
 
         parsed_image, mask = onnx_inference(self.session, input_image, mask_components)
         return parsed_image, mask