Awesome
FER-YOLO-Mamba: Facial Expression Detection and Classification Based on Selective State Space arXiv
📝Introduction📝
This repository is the code implementation of the paper FER-YOLO-Mamba: Facial Expression Detection and Classification Based on Selective State Space.
The current branch has been tested on Linux system, PyTorch 2.0.0 and CUDA 11.7, supports Python Python 3.11.8, and is compatible with CUDA11.7+ versions.
If you find this project helpful, please give us a star ⭐️, your support is our greatest motivation.
📌Installation📌
-
pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1
-
pip install packaging
-
pip install timm==0.4.12
-
pip install causal-conv1d==1.2.0.post2
-
pip install mamba-ssm==1.2.0.post1
📜Other requirements📜:
- Linux System
- NVIDIA GPU
- CUDA 11.7+
✨Acknowledgments✨
We thank the authors of VMamba, MedMamba and YOLOvX for their open-source codes.
💞Citation💞
If you use the code or performance benchmarks of this project in your research, please refer to the following bibtex citation of FER-YOLO-Mamba.
@article{ma2024feryolomamba,
title={FER-YOLO-Mamba: Facial Expression Detection and Classification Based on Selective State Space},
author={Hui Ma and Sen Lei and Turgay Celik and Heng-Chao Li},
journal={arXiv preprint arXiv:2405.01828},
year={2024}
}
Contact Us
If you have any other questions❓, please contact us in time 👬