Skip to content

WeidiXie/Keras-VGGFace2-ResNet50

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

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",
}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages