diff --git a/py/human_parts_ultra.py b/py/human_parts_ultra.py index cbe433c..8e37dca 100644 --- a/py/human_parts_ultra.py +++ b/py/human_parts_ultra.py @@ -73,7 +73,7 @@ class LS_HumanPartsUltra: """ import onnxruntime as ort - model = ort.InferenceSession(model_path, providers=['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider']) + model = ort.InferenceSession(model_path, providers=['CUDAExecutionProvider', 'CPUExecutionProvider', 'TensorrtExecutionProvider']) ret_images = [] ret_masks = [] for img in image: