self-hosted/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-08-08 23:07:06 +02:00
Code Issues Releases Activity

fix(ml): batch axis not being added for recognition model (#12588)

Browse Source 
* fix has_batch_axis

* fix typing
This commit is contained in:
Mert
2024-09-11 23:21:12 -04:00
committed by GitHub
parent fa095c3ca0
commit 22dc9bcebb
2 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,5 +0,0 @@
from app.schemas import ModelSession
def has_batch_axis(session: ModelSession) -> bool:
return not isinstance(session.get_inputs()[0].shape[0], int) or session.get_inputs()[0].shape[0] < 0