fix ort wrapper (#2560)

open-mmlab · Nov 23, 2023 · 8b19586 · 8b19586
1 parent 2882c64
commit 8b19586
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mmdeploy/backend/onnxruntime/init_plugins.py b/mmdeploy/backend/onnxruntime/init_plugins.py
@@ -24,7 +24,7 @@ def get_lib_path() -> str:
         str: The library path to onnxruntime.
     """
     candidates = [
-        '../../lib/libonnxruntime_ops.so*',
+        '../../lib/libonnxruntime.so*',
         '../../lib/onnxruntime.dll',
     ]
     return get_file_path(os.path.dirname(__file__), candidates)
diff --git a/mmdeploy/backend/onnxruntime/wrapper.py b/mmdeploy/backend/onnxruntime/wrapper.py
@@ -48,7 +48,7 @@ def __init__(self,
             # load ort lib before custom ops lib
             lib_path = get_lib_path()
             if osp.exists(lib_path):
-                ctypes.DLL(lib_path)
+                ctypes.CDLL(lib_path)
 
             session_options.register_custom_ops_library(ort_custom_op_path)
             logger.info('Successfully loaded onnxruntime custom ops from '