本模型为以iResNet50为基座的ArcFace模型,在预训练的基础上,使用数据增强的亚洲人脸数据集CASIA_FaceV5进行SFT,旨在提高ArcFace人脸识别模型针对亚洲人脸识别的精确度。
仓库包含了 pth 和 onnx 格式的模型文件,您可以使用pytorch进一步对pth模型文件进行微调,模型结构文件为iresnet.python、arcface.py,本模型使用iResNet50为基座。
onnx文件可以直接用于推理,其中已经包含了模型结构,无序显示定义模型结构。
#Git模型下载
git clone https://huggingface.co/JustinLeee/FaceMind_ArcFace_iResNet50_CASIA_FaceV5