mingzeyang / cr-fiqa Goto Github PK

Arxiv: CR-FIQA

1. In the paper, we employ MS1MV2 as the training dataset for CR-FIQA(L) which can be downloaded from InsightFace (MS1M-ArcFace in DataZoo)
   1. Download MS1MV2 dataset from insightface on strictly follow the licence distribution
2. We use CASIA-WebFace as the training dataset for CR-FIQA(S) which can be downloaded from InsightFace (CASIA in DataZoo)
   1. Download CASIA dataset from insightface on strictly follow the licence distribution
3. Unzip the dataset and place it in the data folder
4. Intall the requirement from requirement.txt
5. pip install -r requirements.txt
6. All code are trained and tested using PyTorch 1.7.1 Details are under (Torch)[https://pytorch.org/get-started/locally/]

Set the following in the config.py

1. config.output to output dir
2. config.network = "iresnet100"
3. config.dataset = "emoreIresNet"
4. Run ./run.sh

Set the following in the config.py

1. config.output to output dir
2. config.network = "iresnet50"
3. config.dataset = "webface"
4. Run ./run.sh

Follow these steps to reproduce the results on XQLFW:

1. Download the aligned XQLFW (please download xqlfw_aligned_112.zip)
2. Note: if you have an unaligned dataset, please do alignment first using Alignment
3. Unzip XQLFW (Folder structure should look like this ./data/XQLFW/xqlfw_aligned_112/)
4. Download also xqlfw_pairs.txt to ./data/XQLFW/xqlfw_pairs.txt
5. Set (in feature_extraction/extract_xqlfw.py) path = "./data/XQLFW" to your XQLFW data folder and outpath = "./data/quality_data" where you want to save the preprocessed data
6. Run python extract_xqlfw.py (it creates the output folder, saves the images in BGR format, creates image_path_list.txt and pair_list.txt)
7. Run evaluation/getQualityScore.py to estimate the quality scores
   1. CR-FIQA(L)
      1. Download the pretrained model
      2. run: python3 evaluation/getQualityScore.py --data_dir "./data/quality_data" --datasets "XQLFW" --model_path "path_to_pretrained_CF_FIQAL_model" --backbone "iresnet100" --model_id "181952" --score_file_name "CRFIQAL.txt"
   2. CR-FIQA(S)
      1. Download the pretrained model
      2. run: python3 evaluation/getQualityScore.py --data_dir "./data/quality_data" --datasets "XQLFW" --model_path "path_to_pretrained_CF_FIQAL_model" --backbone "iresnet50" --model_id "32572" --score_file_name "CRFIQAS.txt"

The quality score of LFW, AgeDB-30, CFP-FP, CALFW, CPLFW can be produced by following these steps:

1. LFW, AgeDB-30, CFP-FP, CALFW, CPLFW are be included in the training dataset folder insightface
2. Set (in extract_bin.py) path = "/data/faces_emore/lfw.bin" to your LFW bin file and outpath = "./data/quality_data" where you want to save the preprocessed data (subfolder will be created)
3. Run python extract_bin.py (it creates the output folder, saves the images in BGR format, creates image_path_list.txt and pair_list.txt)
4. Run evaluation/getQualityScore.py to estimate the quality scores
   1. CR-FIQA(L)
      1. Download the pretrained model
      2. run: python3 evaluation/getQualityScore.py --data_dir "./data/quality_data" --datasets "XQLFW" --model_path "path_to_pretrained_CF_FIQAL_model" --backbone "iresnet100" --model_id "181952" --score_file_name "CRFIQAL.txt" --color_channel "BGR"
   2. CR-FIQA(S)
      1. Download the pretrained model
      2. run: python3 evaluation/getQualityScore.py --data_dir "./data/quality_data" --datasets "XQLFW" --model_path "path_to_pretrained_CF_FIQAL_model" --backbone "iresnet50" --model_id "32572" --score_file_name "CRFIQAS.txt" --color_channel "BGR"

1. Download pretrained model e.g. ElasticFace-Arc, MagFac, CurricularFace or ArcFace
2. Run CUDA_VISIBLE_DEVICES=0 python feature_extraction/extract_emb.py --model_path ./pretrained/ElasticFace --model_id 295672 --dataset_path "./data/quality_data/XQLFW" --modelname "ElasticFaceModel" 2.1 Note: change the path to pretrained model and other arguments according to the evaluated model
3. Run python3 ERC/erc.py (details in ERC/README.md)

If you use any of the code provided in this repository or the models provided, please cite the following paper:

@InProceedings{Boutros_2023_CVPR,
    author    = {Boutros, Fadi and Fang, Meiling and Klemt, Marcel and Fu, Biying and Damer, Naser},
    title     = {CR-FIQA: Face Image Quality Assessment by Learning Sample Relative Classifiability},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2023},
    pages     = {5836-5845}
}

This project is licensed under the terms of the Attribution-NonCommercial 4.0 International (CC BY-NC 4.0) license. Copyright (c) 2021 Fraunhofer Institute for Computer Graphics Research IGD Darmstadt