JustinLeee/FaceMind_ArcFace_iResNet50_CASIA_FaceV5
By JustinLeee · ↓ 0 · ♥ 5 · Updated 2025-03-29
What's in this repo
This page lists the files in JustinLeee/FaceMind_ArcFace_iResNet50_CASIA_FaceV5, including ArcFace_iResNet50_CASIA_FaceV5.onnx.
- 1 ONNX models (.onnx files)
- 7 files, including 2 .py, 1 .onnx, 1 .pth, 1 .md, 1 .json
- Last updated 2025-03-29
ONNX files
| File | Size | Download |
|---|---|---|
| ArcFace_iResNet50_CASIA_FaceV5.onnx | — | Download |
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
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
Model info
Created: 2025-03-29
Updated: 2025-03-29
View on Hugging FaceFiles (7)
Related models
More from JustinLeee
No other models by this author.