ZF/uniface
1
0
Fork 0
mirror of https://github.com/yakhyo/uniface.git synced 2025-12-30 09:02:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Supress the warning and give info about onnx backend

Browse Source
This commit is contained in:
yakhyo
2025-11-26 00:06:39 +09:00
parent 4b8dc2c0f9
commit db7532ecf1
1 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
uniface/onnx_utils.py
View File

@@ -77,10 +77,25 @@ def create_onnx_session(model_path: str, providers: List[str] = None) -> ort.Inf
if providers is None: if providers is None:
providers = get_available_providers() providers = get_available_providers()
# Suppress ONNX Runtime warnings (e.g., CoreML partition warnings)
# Log levels: 0=VERBOSE, 1=INFO, 2=WARNING, 3=ERROR, 4=FATAL
sess_options = ort.SessionOptions()
sess_options.log_severity_level = 3 # Only show ERROR and FATAL
try: try:
session = ort.InferenceSession(model_path, providers=providers) session = ort.InferenceSession(model_path, sess_options=sess_options, providers=providers)
active_provider = session.get_providers()[0] active_provider = session.get_providers()[0]
Logger.debug(f"Session created with provider: {active_provider}") Logger.debug(f"Session created with provider: {active_provider}")
# Show user-friendly message about which provider is being used
provider_names = {
"CoreMLExecutionProvider": "CoreML (Apple Silicon)",
"CUDAExecutionProvider": "CUDA (NVIDIA GPU)",
"CPUExecutionProvider": "CPU",
}
provider_display = provider_names.get(active_provider, active_provider)
print(f"Model loaded ({provider_display})")
return session return session
except Exception as e: except Exception as e:
Logger.error(f"Failed to create ONNX session: {e}", exc_info=True) Logger.error(f"Failed to create ONNX session: {e}", exc_info=True)