KevinMidboe/immich
1
0
Fork 0
mirror of https://github.com/KevinMidboe/immich.git synced 2025-10-29 17:40:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(ml): modularization and styling (#2835)

Browse Source 
* basic refactor and styling

* removed batching

* module entrypoint

* removed unused imports

* model superclass,  model cache now in app state

* fixed cache dir and enforced abstract method

---------

Co-authored-by: Alex Tran <alex.tran1502@gmail.com>
This commit is contained in:
Mert
2023-06-24 23:18:09 -04:00
committed by GitHub
parent 837ad24f58
commit a2f5674bbb
12 changed files with 281 additions and 182 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
machine-learning/app/models/__init__.py Normal file
View File

@@ -0,0 +1,3 @@
from .clip import CLIPSTTextEncoder, CLIPSTVisionEncoder
from .facial_recognition import FaceRecognizer
from .image_classification import ImageClassifier