MT Photos人脸识别API

基于serengil/deepface实现的人脸识别API

模型选择

人脸检测模型

默认的人脸检测模型为retinaface，虽然计算耗时较长，但是最准确；

可通过环境变量 DETECTOR_BACKEND来自定义检测模型；

backends = [
    'opencv',
    'ssd',
    'dlib',
    'mtcnn',
    'retinaface',
    'mediapipe',
    'yolov8',
    'yunet',
    'fastmtcnn',
]
detector_backend = os.getenv("DETECTOR_BACKEND", "retinaface")

人脸特征提取模型

默认的人脸检测模型为Facenet512

可通过环境变量 RECOGNITION_MODEL来自定义特征提取模型；

models = [
    "VGG-Face",
    "Facenet",
    "Facenet512",
    "OpenFace",
    "DeepFace",
    "DeepID",
    "ArcFace",
    "Dlib",
    "SFace",
    "GhostFaceNet",
]
recognition_model = os.getenv("RECOGNITION_MODEL", "Facenet512")

预训练模型

如果容器内下载模型很慢，可以增加 /models 目录映射

-v /host_path:/models

然后将对应的预训练模型放到容器内的/models/.deepface/weights/下；比如/models/.deepface/weights/retinaface.h5

预训练模型下载地址

https://github.com/serengil/deepface_models/releases/tag/v1.0

安装方法

下载镜像

docker pull mtphotos/mt-photos-deepface:latest

创建容器

docker run -i -p 8066:8066 -e API_AUTH_KEY=mt_photos_ai_extra --name mt-photos-deepface --restart="unless-stopped" mtphotos/mt-photos-deepface:latest

打包docker镜像

docker build  . -t mt-photos-deepface:latest

运行docker容器

docker run -i -p 8066:8066 -e API_AUTH_KEY=mt_photos_ai_extra --name mt-photos-deepface --restart="unless-stopped" mt-photos-deepface:latest

下载源码本地运行

安装python 3.8版本
在文件夹下执行pip install -r requirements.txt
复制.env.example生成.env文件，然后修改.env文件内的API_AUTH_KEY
执行 python server.py ，启动服务

看到以下日志，则说明服务已经启动成功

INFO:     Started server process [27336]
INFO:     Waiting for application startup.
INFO:     Application startup complete.
INFO:     Uvicorn running on http://0.0.0.0:8066 (Press CTRL+C to quit)

API

/check

检测服务是否可用，及api-key是否正确

curl --location --request POST 'http://127.0.0.1:8000/check' \
--header 'api-key: api_key'

response:

{
  "result": "pass"
}

/represent

curl --location --request POST 'http://127.0.0.1:8000/represent' \
--header 'api-key: api_key' \
--form 'file=@"/path_to_file/test.jpg"'