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

fix typo and add a propper var name

Browse Source
This commit is contained in:
yoni13
2025-01-18 10:46:57 +08:00
parent 3634ae1f5b
commit f5de3de163
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
machine-learning/rknn/rknnpool.py
+4 -4
View File
@@ -31,7 +31,7 @@ except (FileNotFoundError, ImportError):
soc_name = None
def initRKNN(rknnModel="./rknnModel/yolov5s.rknn", id=0):
def initRKNN(rknnModel, id) -> RKNNLite:
if not is_available:
raise RuntimeError("rknn is not available!")
rknn_lite = RKNNLite()
@@ -59,7 +59,7 @@ def initRKNN(rknnModel="./rknnModel/yolov5s.rknn", id=0):
return rknn_lite
def initRKNNs(rknnModel="./rknnModel/yolov5s.rknn", tpes=1):
def initRKNNs(rknnModel, tpes) -> list[RKNNLite]:
rknn_list = []
for i in range(tpes):
rknn_list.append(initRKNN(rknnModel, i % 3))
@@ -75,8 +75,8 @@ class RknnPoolExecutor:
self.func = func
self.num = 0
def put(self, frame) -> None:
self.queue.put(self.pool.submit(self.func, self.rknn_pool[self.num % self.tpes], frame))
def put(self, inputs) -> None:
self.queue.put(self.pool.submit(self.func, self.rknn_pool[self.num % self.tpes], inputs))
self.num += 1
def get(self) -> list[list[NDArray[np.float32]], bool]: