GitHub - HaipengMing/SLM-AEA

A Simple Reproduction of Our Method

Requirement

Python 3.9, Torch 1.11.0 and torchvision 0.12.0 are recommended.
Data

Download basic emotions dataset of RAF-DB. We have used pickle to make a list of filenames, and it have a structure as following. You can also make localized pickle files using our provided "pickleData.py".

 ./datasets/aligned_RAF/
	     train_00001_aligned.jpg
             test_0001_aligned.jpg
             ...

   python main.py

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
datasets		datasets
models		models
README.md		README.md
fer_dataset.py		fer_dataset.py
main.py		main.py
tools.py		tools.py