-
PyTorch人脸识别器VGGFace2-pytorch,基于\'VGGFace2\'
资源介绍
基于“ VGGFace2:用于识别跨姿势和年龄的面部表情的数据集”的PyTorch面部表情识别器。
此仓库实现了培训和测试模型,并基于VGGFace2 [1]的模型构建了特征提取器。
是从[1]的作者提供的转换而来的。
数据集
要下载VGGFace2数据集,请参见。
预处理图像
在将脸部图像馈送到该脸部识别器( demo.py )之前,应先检测脸部并从图像中裁剪出脸部。
有几种基于MTCNN的面部检测程序[3]。
PyTorch版本:
MXNet版本:
Matlab版本:
预训练模型
以下是从[1]的作者提供的模型转换而来的PyTorch模型。
arch_type
下载链接
resnet50_ft
senet50_ft
resnet50_scratch
senet50_scratch
提取特征
用法:
python demo.py extract < options>