set log level

This commit is contained in:
mertalev 2025-03-14 23:37:39 -04:00
parent bc9bec8673
commit 76eb285204
No known key found for this signature in database
GPG Key ID: 8AC1FB7DEC668BCC

View File

@ -2,6 +2,7 @@
# Following Apache License 2.0
from concurrent.futures import Future, ThreadPoolExecutor
import logging
from pathlib import Path
from queue import Queue
from typing import Callable
@ -41,6 +42,7 @@ def init_rknn(model_path: str) -> "RKNNLite":
if not is_available:
raise RuntimeError("rknn is not available!")
rknn_lite = RKNNLite()
rknn_lite.rknn_log.logger.setLevel(logging.ERROR)
ret = rknn_lite.load_rknn(model_path)
if ret != 0:
raise RuntimeError("Failed to load RKNN model")