immich/machine-learning
Mert a2f5674bbb
refactor(ml): modularization and styling (#2835)
* 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>
2023-06-24 22:18:09 -05:00
..
app refactor(ml): modularization and styling (#2835) 2023-06-24 22:18:09 -05:00
.dockerignore feat: facial recognition (#2180) 2023-05-17 12:07:17 -05:00
.gitignore feat: facial recognition (#2180) 2023-05-17 12:07:17 -05:00
Dockerfile refactor(ml): modularization and styling (#2835) 2023-06-24 22:18:09 -05:00
poetry.lock feat(ml) backend takes image over HTTP (#2783) 2023-06-17 22:49:19 -05:00
pyproject.toml fix(deps): install poetry in pump workflow (#2938) 2023-06-24 22:03:50 -05:00
README.md chore(ml): updated dockerfile, added typing, packaging (#2642) 2023-06-05 09:40:48 -05:00

Immich Machine Learning

  • Image classification
  • CLIP embeddings
  • Facial recognition

Setup

This project uses Poetry, so be sure to install it first. Running poetry install --no-root --with dev will install everything you need in an isolated virtual environment.

To add or remove dependencies, you can use the commands poetry add $PACKAGE_NAME and poetry remove $PACKAGE_NAME, respectively. Be sure to commit the poetry.lock and pyproject.toml files to reflect any changes in dependencies.