Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-07-09 03:04:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

Handling Import and file not found Error for non-arm devices.

Browse Source
This commit is contained in:
yoni13 2025-01-11 15:19:53 +08:00
parent 66004e3b83
commit d10147f478
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2,12 +2,13 @@
# Following Apache License 2.0 # Following Apache License 2.0
from queue import Queue from queue import Queue
from rknnlite.api import RKNNLite
from concurrent.futures import ThreadPoolExecutor, as_completed from concurrent.futures import ThreadPoolExecutor, as_completed
supported_socs = ["rk3566","rk3568","rk3588"] supported_socs = ["rk3566","rk3568","rk3588"]
with open('/proc/device-tree/compatible') as f: try:
from rknn.api import RKNNLite
with open('/proc/device-tree/compatible') as f:
# Keep in mind that this is not in container by default. # Keep in mind that this is not in container by default.
# and this way of checking can't check if the rknpu driver is running or not. # and this way of checking can't check if the rknpu driver is running or not.
device_compatible_str = f.read() device_compatible_str = f.read()
@ -17,7 +18,8 @@ with open('/proc/device-tree/compatible') as f:
break break
else: else:
is_available = False is_available = False
except (FileNotFoundError, ImportError):
is_available = False
def initRKNN(rknnModel="./rknnModel/yolov5s.rknn", id=0): def initRKNN(rknnModel="./rknnModel/yolov5s.rknn", id=0):