Update README.md

f8900d1 verified about 1 year ago

1.59 kB

frameworks:
  - Pytorch
license: apache-2.0
tasks:
  - face-recognition
model-type:
  - ArcFace_iResNet50
domain:
  - cv
language:
  - zh
  - en
base_model_relation: finetune
metrics:
  - accuracy
tags:
  - ArcFace
  - CASIA_FaceV5
  - 亚洲人脸
  - 中国人脸
  - fine-tuned
datasets:
  - JustinLeee/Cleaned_Augmented_CASIA_FaceV5
pipeline_tag: image-classification

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