This is the official PyTorch implementation of CVPR 2024 paper (LAFS: Landmark-based Facial Self-supervised Learning for Face Recognition).
Our code is partly borrowed from DINO (https://github.com/facebookresearch/dino) and Insightface(https://github.com/deepinsight/insightface).
In this work we focus on learning facial representations that can be adapted to train effective face recognition models, particularly in the absence of labels. Firstly, compared with existing labelled face datasets, a vastly larger magnitude of unlabeled faces exists in the real world. We explore the learning strategy of these unlabeled facial images through self-supervised pretraining to transfer generalized face recognition performance. Moreover, motivated by one recent finding, that is, the face saliency area is critical for face recognition, in contrast to utilizing random cropped blocks of images for constructing augmentations in pretraining, we utilizcd e patches localized by extracted facial landmarks. This enables our method - namely Landmark-based Facial Self-supervised learning (LAFS), to learn key representation that is more critical for face recognition. We also incorporate two landmark-specific augmentations which introduce more diversity of landmark information to further regularize the learning. With learned landmark-based facial representations, we further adapt the representation for face recognition with regularization mitigating variations in landmark positions. Our method achieves significant improvement over the state-of-the-art on multiple face recognition benchmarks, especially on more challenging few-shot scenarios.
@InProceedings{Sun_2024_CVPR,
author = {Sun, Zhonglin and Feng, Chen and Patras, Ioannis and Tzimiropoulos, Georgios},
title = {LAFS: Landmark-based Facial Self-supervised Learning for Face Recognition},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2024}
}
Please consider cite our paper and star the repo if you find this repo useful.
- LAFS Pretraining scripts
- DINO-face Pretraining scripts
- Checkpoints
- Finetuning scripts
- IJB datasets evaluation code
Please stay tuned for more updates.
- Pytorch Version
torch==1.8.1+cu111; torchvision==0.9.1+cu111
- Dataset
- MS1MV3 -- Please download from InsightFace(https://github.com/deepinsight/insightface/tree/master/recognition/_datasets_). Note that MS1MV3 use bgr order, and WebFace4M use rgb order.
- WebFace4m
- SSL Pretraining Command Before you start self-supervised pretraining, please use the landmark weight trained on MS1MV3 or WebFace4M, and specify in --landmark_path:
- Part-fViT MS1MV3 (Performance on IJB-C: TAR@FAR=1e-4 97.29).
- Part-fViT WebFace4M
python -m torch.distributed.launch --nproc_per_node=2 lafs_train.py
Note on 2A100 (40GB), the total pretraining training time would be around 2-3 days.
- Supervised finetuning Command The training setting difference between the MS1MV3 and WebFace4M is that MS1MV3 use a stronger mixup probability of 0.2 and rand augmentation with magnitude of 2, while WebFace4M use 0.1 for mixup and rand augmentation magnitude.
And please note the colour channel of these datasets, i.e. MS1MV3 use brg order.
Before you run the following command, please change the dataset path --dataset_path, SSL pretrained model --model_dir, and model from stage 1 --pretrain_path.
If you want to run the model with flip augmentation only, please disable the mixup and augmentations, by setting random_resizecrop, rand_au to False in the FaceDataset function, and set mixup-prob to 0.0
python -m torch.distributed.launch --nproc_per_node=2 --nnodes=1 --node_rank=0 --master_port 47771 train_largescale.py
Please let me know if there is any problem in using this code as the cleaned code hasn't been tested. I will keep updating the usage of the code, thank you!
This project is licensed under the terms of the MIT license.