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

make these functions snake case.

Browse Source
This commit is contained in:
yoni13 2025-01-18 11:03:54 +08:00
parent f5de3de163
commit be76857ae6
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -31,7 +31,7 @@ except (FileNotFoundError, ImportError):
soc_name = None
def initRKNN(rknnModel, id) -> RKNNLite:
def init_rknn(rknnModel, id) -> RKNNLite:
if not is_available:
raise RuntimeError("rknn is not available!")
rknn_lite = RKNNLite()
@ -59,10 +59,10 @@ def initRKNN(rknnModel, id) -> RKNNLite:
return rknn_lite
def initRKNNs(rknnModel, tpes) -> list[RKNNLite]:
def init_rknns(rknnModel, tpes) -> list[RKNNLite]:
rknn_list = []
for i in range(tpes):
rknn_list.append(initRKNN(rknnModel, i % 3))
rknn_list.append(init_rknn(rknnModel, i % 3))
return rknn_list
@ -70,7 +70,7 @@ class RknnPoolExecutor:
def __init__(self, rknnModel: str, tpes: int, func):
self.tpes = tpes
self.queue = Queue()
self.rknn_pool = initRKNNs(rknnModel, tpes)
self.rknn_pool = init_rknns(rknnModel, tpes)
self.pool = ThreadPoolExecutor(max_workers=tpes)
self.func = func
self.num = 0