ONNX 模型库
返回模型

说明文档

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

JustinLeee/FaceMind_ArcFace_iResNet50_CASIA_FaceV5

作者 JustinLeee

image-classification
↓ 0 ♥ 5

创建时间: 2025-03-29 08:38:22+00:00

更新时间: 2025-03-29 13:48:35+00:00

在 Hugging Face 上查看

文件 (7)

.gitattributes
ArcFace_iResNet50_CASIA_FaceV5.onnx ONNX
ArcFace_iResNet50_CASIA_FaceV5.pth
README.md
arcface.py
configuration.json
iresnet.py