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
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train JustinLeee/FaceMind_ArcFace_iResNet50_CASIA_FaceV5