Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-06-05 14:44:41 -04:00
Code Issues Packages Projects Releases Wiki Activity

fixed some bugs

Browse Source
This commit is contained in:
yoni13 2025-01-14 10:38:45 +00:00
parent 4b0f93cf6a
commit 8b80d034cb
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
machine-learning/app/sessions/rknn.py
View File

@ -13,8 +13,8 @@ from rknn.rknnpool import rknnPoolExecutor, soc_name
from ..config import log, settings from ..config import log, settings
def runInfrence(rknn_lite: Any, input: NDArray[np.float32]) -> list[NDArray[np.float32]]: def runInfrence(rknn_lite: Any, input: list[NDArray[np.float32]]) -> list[NDArray[np.float32]]:
outputs: list[NDArray[np.float32]] = rknn_lite.inference(inputs=[input], data_format="nchw") outputs: list[NDArray[np.float32]] = rknn_lite.inference(inputs=input, data_format="nchw")
return outputs return outputs
@ -66,7 +66,7 @@ class RknnSession:
input_feed: dict[str, NDArray[np.float32]] | dict[str, NDArray[np.int32]], input_feed: dict[str, NDArray[np.float32]] | dict[str, NDArray[np.int32]],
run_options: Any = None, run_options: Any = None,
) -> list[NDArray[np.float32]]: ) -> list[NDArray[np.float32]]:
input_data: NDArray[np.float32] = np.ascontiguousarray(list(input_feed.values())[0], dtype=np.float32) input_data: list[NDArray[np.float32]] = [np.ascontiguousarray(v) for v in input_feed.values()]
self.rknnpool.put(input_data) self.rknnpool.put(input_data)
outputs: list[NDArray[np.float32]] = self.rknnpool.get()[0] outputs: list[NDArray[np.float32]] = self.rknnpool.get()
return outputs return outputs

6
machine-learning/rknn/rknnpool.py
View File

@ -23,7 +23,7 @@ try:
else: else:
is_available = False is_available = False
soc_name = None soc_name = None
is_available = os.path.exists("/sys/kernel/debug/rknpu/load") is_available = is_available and os.path.exists("/sys/kernel/debug/rknpu/load")
except (FileNotFoundError, ImportError): except (FileNotFoundError, ImportError):
is_available = False is_available = False
soc_name = None soc_name = None
@ -79,9 +79,9 @@ class rknnPoolExecutor:
def get(self) -> list[list[NDArray[np.float32]], bool]: def get(self) -> list[list[NDArray[np.float32]], bool]:
if self.queue.empty(): if self.queue.empty():
return None, False return None
fut = self.queue.get() fut = self.queue.get()
return fut.result(), True return fut.result()
def release(self) -> None: def release(self) -> None:
self.pool.shutdown() self.pool.shutdown()