hangyu94 / Ada-CM

“Towards Semi-Supervised Deep Facial Expression Recognition with An Adaptive Confidence Margin”, CVPR 2022.

Ada-CM

This is the official PyTorch implementation of our CVPR 2022 paper:

Towards Semi-Supervised Deep Facial Expression Recognition with An Adaptive Confidence Margin
Hangyu Li, Nannan Wang*, Xi Yang, Xiaoyu Wang, Xinbo Gao
In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2022

Update

• (2022/04/28) We upload the main files for training on RAF-DB.
• (2022/05/10) We provide the pre-trained ResNet-18 on MS-Celeb-1M. Please put it into ./models.

Requirements

We test the codes in the following environments:

• Python==3.6.9
• PyTorch==1.3.0
• Torchvision==0.4.1

Dataset setup

Please download the dataset from RAF-DB website and change the root to your path. The splits of different-class labeled data are listed below. For example, for the case of 100 labels, the labeled training set consists of 10 faces annotated with fear and 15 faces annotated with other expressions. More details could be found in the supplementary material, which is available at CVF website.

	100 labels	400 labels	1000 labels	2000 labels	4000 labels
Fear	10	40	100	200	250
Others	15	60	150	300	625

Getting Started

To train on RAF-DB:

python main.py

Citation

If you find our work useful in your research, please consider citing:

@inproceedings{li2022adacm,
  title={Towards Semi-Supervised Deep Facial Expression Recognition with An Adaptive Confidence Margin},
  author={Li, Hangyu and Wang, Nannan and Yang, Xi and Wang, Xiaoyu and Gao, Xinbo},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
  month={June},
  year={2022},
  pages={4166-4175}
}

Note

If you have any questions, please contact me. Email: [e-mail]