概述

本项目可以自动识别图像中的公众人物, 并提供脸部坐标和置信度得分.

目前可识别 5065 个公众人物, 包括歌星, 演员, 运动员和政要等, 支持名单在不断更新中.

本项目的人脸检测算法和模型来自 Pytorch_Retinaface , 人脸特征提取模型来自 InsightFace_Pytorch .

安装

git clone https://github.com/quarrying/quarrying-celebrity-id.git
cd quarrying-celebrity-id
conda create -n celebid python=3.8 -y
conda activate celebid
pip install -r requirements.txt

用法

参考 demo.py

数值是欧氏距离, 越小越相似.

ChangeLog

• 20211018 更新名人人脸特征库, 现支持 5065 人.
• 20211012 更新名人人脸特征库, 现支持 4846 人.
• 20211011 更新名人人脸特征库, 现支持 4801 人.
• 20210615 发布开源项目, 支持 4692 个名人人脸识别.

参考

• https://github.com/biubug6/Pytorch_Retinaface
• https://github.com/TreB1eN/InsightFace_Pytorch