cjones/immich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

black app export

Browse Source
This commit is contained in:
yoni13
2025-01-14 18:40:28 +08:00
parent 8b80d034cb
commit 5244ed6d4d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
machine-learning/app/sessions/rknn.py
+1 -1
View File
@@ -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: list[NDArray[np.float32]] = [np.ascontiguousarray(v) for v in input_feed.values()] 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() outputs: list[NDArray[np.float32]] = self.rknnpool.get()
return outputs return outputs