Skip to content

Official repository for the OCFR 2022: Competition on Occluded Face Recognition From Synthetically Generated Structure-Aware Occlusions at 2022 International Joint Conference on Biometrics (IJCB 2022).

License

NetoPedro/OCFR-2022

Repository files navigation

OCFR-2022

Official repository for the OCFR 2022: Competition on Occluded Face Recognition From Synthetically Generated Structure-Aware Occlusions at 2022 International Joint Conference on Biometrics (IJCB 2022).

The paper can be viewed at: arXiv and IEEE

Data

All the data used for the competition is available for download as a benchmark dataset.

Images for the 8 different protocols are provided in the zip file, within individual folders. The list of images per protocol is provided in the form of Reference_Probe_list.txt, where Reference and Probe are replaced by the occlusion strategy O[1-7] or C (from Clear).

The issame list is equal for all protocols, as well as the pairs list. Please be aware that the same image might appear non-occluded as a reference and occluded as a probe (while in different pairs). Hence, we provide a duplicate of all images with a renaming indicating if it is used as probe or not.

We provide two images: the occlusion (a mask-like image where all the pixels except the occlusion pixels are set to 0), and an occlusion mask (similar to the previous, but occlusion pixels are set to 0 and the others are set to 255). To construct an occluded image, you should have the respective LFW image, and use an equivalent python command img[mask==0] = occlusion[mask==0]

Download : gdrive

Generate new data

python3 align_db.py --input-dir=input_path --output-dir=output_path --image-size=112

Replicating protocols

1

def select_occlusion_type():
    return np.random.choice([1,2,4]) 

2

def select_occlusion_type():
    return np.random.choice([5,6,7,10,11]) 

3

def select_occlusion_type():
    return np.random.choice([1,2,4,5,6,7,10,11]) 

4

def select_occlusion_type():
    return np.random.choice([1,2,4,6,7,10,11]) 

5

def select_occlusion_type():
    return np.random.choice([8,9]) 

6

def select_occlusion_type():
    return np.random.choice([5,6,7,8,9,10,11]) 

7

def select_occlusion_type():
    return np.random.choice([1,2,4,5,6,7,8,9,10,11]) 

Evaluation Script

Please also find a sample of an evaluation script in the "evaluation.py" file.

Example of the read function

  • Non-occluded path is in the form: XXX/lfwA_u/identity/YYY.jpg
  • Occluded paths (in the csv files) are in the form: XXX/lfwAO[1-7]_u/identity/YYY.jpg
  • Occlusions paths are in the form: XXX/occlusions_[1-7]/identity/YYY_occlusion.jpg
  • Mask paths are in the form: XXX/occlusions_[1-7]/identity/YYY_occlusion_mask.jpg
def read(self,image_path):
    if "lfwA_u" in image_path: #if the image is non-occluded.
        img = cv2.imread(image_path)
    else: 
        mask_path = image_path.replace("lfwAO","occlusions_").replace("_u/","/") 
        occlusion_path = mask_path.replace(".jpg","_occlusion.jpg") #get the occlusion path
        mask_path = mask_path.replace(".jpg","_occlusion_mask.jpg") #get the mask_path
        occlusion = cv2.imread(occlusion_path) # read the occlusion
        mask = cv2.imread(mask_path) # read the mask
        image_path = image_path.split("/") #redirect the path from the occluded to the non-occluded image
        image_path[4] = "lfwA_u" 
        img = cv2.imread(os.path.join(*image_path)) #read image
        img[mask==0] = occlusion[mask==0] # merge occlusion
    return img

The code above can be simplified if the CSV files are directly modified to replace "lfwAO" with "occlusions_" and "_u/" with "/".

Including new occluders

We will release the code to generate the occluded images soon. Just by adding a new occluder image and its information in the "occluders/occluders.csv" you can use the script to create a new benchmark dataset. Hence, this dataset can be extended overtime with contributions from the community.

Moreover, these occlusions can be easly applied to other base datasets. For instance, AgeDB-30, MS1MV2, etc.

Acknowledgement

The code was extended from the initial code of Self-restrained-Triplet-Loss.

Citation

If you use our code or data in your research, please cite with:

@inproceedings{neto2022ocfr,
  title={OCFR 2022: Competition on occluded face recognition from synthetically generated structure-aware occlusions},
  author={Neto, Pedro C and Boutros, Fadi and Pinto, Joao Ribeiro and Damer, Naser and Sequeira, Ana F and Cardoso, Jaime S and Bengherabi, Messaoud and Bousnat, Abderaouf and Boucheta, Sana and Hebbadj, Nesrine and others},
  booktitle={2022 IEEE International Joint Conference on Biometrics (IJCB)},
  pages={1--9},
  year={2022},
  organization={IEEE}
}

and

@TechReport{LFWTech,
  author =       {Gary B. Huang and Manu Ramesh and Tamara Berg and 
                  Erik Learned-Miller},
  title =        {Labeled Faces in the Wild: A Database for Studying 
                  Face Recognition in Unconstrained Environments},
  institution =  {University of Massachusetts, Amherst},
  year =         2007,
  number =       {07-49},
  month =        {October}}

TODO

  • Link to the preprint
  • Citation to arXiv
  • Link to the published paper
  • Citation published version
  • Evaluation script
  • Upload occluders
  • Script to generate the data

License

Attribution-NonCommercial-ShareAlike 4.0 International
This project is licensed under the terms of the Attribution-NonCommercial-ShareAlike 4.0
International (CC BY-NC-SA 4.0) license. 
Copyright (c) 2022 Instituto de Engenharia de Sistemas e Computadores, Tecnologia e Ciência

About

Official repository for the OCFR 2022: Competition on Occluded Face Recognition From Synthetically Generated Structure-Aware Occlusions at 2022 International Joint Conference on Biometrics (IJCB 2022).

Topics

Resources

License

Stars

Watchers

Forks

Languages