elestio4test/immich-machine-learning Docker 镜像 - 轩辕镜像 | Docker 镜像高效稳定拉取服务

Immich机器学习镜像

镜像概述和主要用途

Immich机器学习镜像是Immich应用的关键组件，专门提供图像相关的机器学习功能，包括图像分类、CLIP嵌入生成和面部识别，旨在增强Immich的媒体管理能力，提升用户对照片和视频的组织与检索效率。

核心功能和特性

• 图像分类：自动识别图像内容并进行分类
• CLIP嵌入：生成图像的CLIP嵌入向量，支持语义搜索和相似性比较
• 面部识别：检测和识别图像中的人脸，实现人物分组和标记

使用场景和适用范围

适用于Immich应用环境中，需要对大量照片和视频进行智能处理的场景，如家庭媒体库管理、个人照片分类、人脸自动标记等，帮助用户更高效地组织和检索媒体文件。

使用方法和配置说明

环境准备

该项目使用Poetry进行依赖管理，使用前需先安装Poetry。

安装依赖

在隔离的虚拟环境中安装所需依赖：

bash
poetry install --no-root --with dev

依赖管理

• 添加依赖：poetry add $PACKAGE_NAME
• 移除依赖：poetry remove $PACKAGE_NAME
• 依赖变更后，需提交poetry.lock和pyproject.toml文件以反映更改

负载测试

使用Locust工具（通过提供的locustfile.py）测量推理吞吐量和延迟。Locust通过查询模型端点并聚合统计数据工作，因此需要先部署应用。

启动Locust

bash
locust --web-host 127.0.0.1

然后在浏览器中打开localhost:8089访问Locust UI，可在UI中更改模型或调整分数阈值等选项。

并发设置说明

在Locust术语中，并发通过"用户数"衡量，每个用户一次执行一个任务。要实现特定的端点并发量，需将端点数量乘以所需的每个端点并发数。例如，若有3个端点，每个端点需同时处理8个请求，则应设置用户数为24（3×8）。

更多Locust使用信息请参考Locust文档。