Awesome
VGGFace2 Extension
This repo contains a Keras implementation of the paper,
VGGFace2: A dataset for recognising faces across pose and age (Cao et al., FG 2018).
Dependencies
Data
The dataset used for the experiments are
- VGGFace2 [1]
Model
Keras Model (https://drive.google.com/file/d/1AHVpuB24lKAqNyRRjhX7ABlEor6ByZlS/view?usp=sharing),
Note:
This model is trained with a slightly different tight crops, but I have also tested on the tight crops (as we did in the paper), and am able to get similar results (on both IJBB and IJBC).
Dataset | Feat dim | Pretrain | TAR@FAR = 1e-5 | TAR@FAR = 1e-4 | TAR@FAR = 1e-3 | TAR@FAR = 1e-2 | TAR@FAR = 1e-1 |
---|---|---|---|---|---|---|---|
IJBB | 512 | N | 0.64 | 0.78 | 0.88 | 0.94 | 0.98 |
IJBC | 512 | N | 0.72 | 0.82 | 0.90 | 0.95 | 0.98 |
Testing the model
To test a specific model on the IJB dataset, for example, the model trained with ResNet50 trained by sgd with softmax, and feature dimension 512
- python predict.py --net resnet50 --batch_size 64 --gpu 2 --loss softmax --aggregation avg --resume ../model/resnet50_softmax_dim512/weights.h5 --feature_dim 512
Citation
@InProceedings{Cao18,
author = "Q. Cao, L. Shen, W. Xie, O. M. Parkhi, A. Zisserman ",
title = "VGGFace2: A dataset for recognising face across pose and age",
booktitle = "International Conference on Automatic Face and Gesture Recognition, 2018.",
year = "2018",
}