immich-app/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2026-07-25 14:00:45 +03:00
Code Issues 783 Packages Projects Releases 22 Wiki Activity

[PR #4952] [MERGED] feat(ml): add face models #10573

New Issue
Closed
opened 2026-02-05 14:24:21 +03:00 by OVERLORD · 0 comments
OVERLORD commented 2026-02-05 14:24:21 +03:00
Owner
Copy Link

📋 Pull Request Information

Original PR: https://github.com/immich-app/immich/pull/4952
Author: @mertalev
Created: 11/11/2023
Status: ✅ Merged
Merged: 11/12/2023
Merged by: @alextran1502

Base: main ← Head: feat/ml-antelopev2

📝 Commits (1)

📊 Changes

8 files changed (+101 additions, -94 deletions)

View changed files

📝 machine-learning/app/config.py (+9 -1)
📝 machine-learning/app/models/__init__.py (+6 -4)
📝 machine-learning/app/models/base.py (+8 -3)
📝 machine-learning/app/models/clip.py (+2 -60)
➕ machine-learning/app/models/constants.py (+57 -0)
📝 machine-learning/app/models/facial_recognition.py (+14 -23)
📝 machine-learning/app/test_main.py (+2 -2)
📝 web/src/lib/components/admin-page/settings/machine-learning-settings/machine-learning-settings.svelte (+3 -1)

📄 Description

Description

This PR adds antelopev2 and buffalo_m as supported facial recognition models. antelopev2 has the same detection model as buffalo_l, but a roughly 50% larger recognition model. buffalo_m has the same recognition model as buffalo_l, but a smaller detection model.

Since these models aren't available through their Python API, facial recognition models are now hosted and downloaded from Hugging Face.

Fixes #4924

How Has This Been Tested?

Tested downloading and running each facial recognition model with local POST requests. Image classification and CLIP models continue to work as normal as well.

🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.

## 📋 Pull Request Information **Original PR:** https://github.com/immich-app/immich/pull/4952 **Author:** [@mertalev](https://github.com/mertalev) **Created:** 11/11/2023 **Status:** ✅ Merged **Merged:** 11/12/2023 **Merged by:** [@alextran1502](https://github.com/alextran1502) **Base:** `main` ← **Head:** `feat/ml-antelopev2` --- ### 📝 Commits (1) - [`057179a`](https://github.com/immich-app/immich/commit/057179aa0cde57e5606a866b299d938c6c9d48ba) add face models ### 📊 Changes **8 files changed** (+101 additions, -94 deletions) <details> <summary>View changed files</summary> 📝 `machine-learning/app/config.py` (+9 -1) 📝 `machine-learning/app/models/__init__.py` (+6 -4) 📝 `machine-learning/app/models/base.py` (+8 -3) 📝 `machine-learning/app/models/clip.py` (+2 -60) ➕ `machine-learning/app/models/constants.py` (+57 -0) 📝 `machine-learning/app/models/facial_recognition.py` (+14 -23) 📝 `machine-learning/app/test_main.py` (+2 -2) 📝 `web/src/lib/components/admin-page/settings/machine-learning-settings/machine-learning-settings.svelte` (+3 -1) </details> ### 📄 Description ## Description This PR adds `antelopev2` and `buffalo_m` as supported facial recognition models. `antelopev2` has the same detection model as `buffalo_l`, but a roughly 50% larger recognition model. `buffalo_m` has the same recognition model as `buffalo_l`, but a smaller detection model. Since these models aren't available through their Python API, facial recognition models are now hosted and downloaded from Hugging Face. Fixes #4924 ## How Has This Been Tested? Tested downloading and running each facial recognition model with local POST requests. Image classification and CLIP models continue to work as normal as well. --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>
OVERLORD added the pull-request label 2026-02-05 14:24:21 +03:00
Sign in to join this conversation.
No Branch/Tag Specified
Branches Tags
Labels
Clear labels
accessibility
changelog:enhancement
changelog:security
changelog:skip
changelog:translation
cli
date-time
dependencies
documentation
external-library
format
good first issue
mobile-beta
mobile-beta
mobile-beta
needs-answer
nice to have
pull-request
Mirrored from GitHub Pull Request
 sharing
tech-debt
📱mobile
🖥️web
🗄️server
🧠machine-learning
No Label pull-request
Milestone
No items
No Milestone
Projects
Clear projects
No project
Assignees
Clear assignees
OVERLORD
No Assignees
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: immich-app/immich#10573
Delete Branch "%!s()"

Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?