FaceMind_ArcFace:针对亚洲人脸 SFT 的 ArcFace 人脸识别模型
本模型为以iResNet50为基座的ArcFace模型,在预训练的基础上,使用数据增强的亚洲人脸数据集CASIA_FaceV5进行SFT,旨在提高ArcFace人脸识别模型针对亚洲人脸识别的精确度。
Reference
本模型用于我个人的人脸识别系统 FaceMind:https://github.com/Justin-ljw/FaceMind
SFT使用的数据集(我个人清洗并数据增强的CASIA_FaceV5):https://modelscope.cn/datasets/JustinLeee/Cleaned_Augmented_CASIA_FaceV5
预训练模型和训练代码来自GitHub大佬:https://github.com/bubbliiiing/arcface-pytorch
仓库说明
仓库包含了 pth 和 onnx 格式的模型文件,您可以使用pytorch进一步对pth模型文件进行微调,模型结构文件为iresnet.python、arcface.py,本模型使用iResNet50为基座。 onnx文件可以直接用于推理,其中已经包含了模型结构,无序显示定义模型结构。
快速开始
模型文件和权重,可浏览“模型文件”页面获取。您可以通过如下git clone命令来下载模型
Git下载
#Git模型下载
git clone https://huggingface.co/JustinLeee/FaceMind_ArcFace_iResNet50_CASIA_FaceV5