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

Update rknn.py

Browse Source
This commit is contained in:
Yoni Yang 2025-01-10 20:20:21 +08:00 committed by GitHub
parent 8608b9c6c5
commit 9bc3e5b2e2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -58,10 +58,7 @@ class RknnSession:
# log.info(f"Running inference on RKNN model")
outputs = self.rknn.inference(inputs=[input_data])
if not outputs:
outputs = self.rknn.inference(inputs=[input_data], data_format='nchw')
outputs = self.rknn.inference(inputs=[input_data], data_format='nchw')
# log.info("inputs:")
# log.info(input_data)