Skip to content

Prediction of Evoked Expression from Videos with Temporal Position Fusion

License

Notifications You must be signed in to change notification settings

th2l/EvokedExpr-TPF

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Prediction of Evoked Expression from Videos with Temporal Position Fusion

Updated code and model checkpoints for EEV 2021 dataset and LIRIS-ACCEDE MediaEval 2018. Paper submitted.

Feature extraction

  • Dockerfile: create docker image for feature extraction with TensorFlow Hub.
  • Adjust line 20 in feature_extractor.py and line 19 in utils.py, dataset_root_path variable to correct folder on your computer. This folder need to contained videos in EEV with 3 separated folders (train, val, test).
  • The structure of dataset_root_path folder as in my code. Note that eev folder is from https://github.com/google-research-datasets/eev
/mnt/Work/Dataset/EEV/
└───dataset
│   └───train
│       │   __18H35fPo8.mp4
│       │   _-6yhPXr_Hw.mp4
│       │   ...
│   └───val
│       │   _-91_iXATY8.mp4
│       │   _-zO8Gg2Kxw.mp4
│       │   ...
│   └───test
│       │   _EARmKIxjxQ.mp4
│       │   _FJgj1e_xo8.mp4
│       │   ...
│   
└───eev
    │   test.csv
    │   train.csv
    │   val.csv

Network training

  • environment.yml: contain list of packages (using anaconda) for network training conda env create -f environment.yml
  • Adjust line 25 in main.py, root_path parameter to correct folder that contain feature extracted from above.
  • Adjust line 184 in models.py, dataset_root_path to the same value with root_path in main.py
  • The structure of root_path folder as in my code. Note that eev folder is from https://github.com/google-research-datasets/eev
/mnt/sXProject/EvokedExpression/
└───dataset
│   └───features_v2
│       └───train
│           │   __18H35fPo8.pt
│           │   _-6yhPXr_Hw.pt
│           │   ...
│       └───val
│           │   _-91_iXATY8.pt
│           │   _-zO8Gg2Kxw.pt
│           │   ...
│       └───test
│           │   _EARmKIxjxQ.pt
│           │   _FJgj1e_xo8.pt
│           │   ...
│   
└───eev
    │   test.csv
    │   train.csv
    │   val.csv

Please cite related papers (dataset, backbone) and our paper https://www.sciencedirect.com/science/article/pii/S016786552300199X

@article{HUYNH2023245,
    title = {Prediction of evoked expression from videos with temporal position fusion},
    journal = {Pattern Recognition Letters},
    volume = {172},
    pages = {245-251},
    year = {2023},
    issn = {0167-8655},
    doi = {https://doi.org/10.1016/j.patrec.2023.07.002},
    author = {Van Thong Huynh and Hyung-Jeong Yang and Guee-Sang Lee and Soo-Hyung Kim},
}