Implementation of FOMM with AWP and MRAA with AWP.
CASE | Driving | FOMM | FOMM w/ EWP | FOMM w/ AWP | MRAA | MRAA w/ AWP |
---|---|---|---|---|---|---|
1 | ||||||
2 | ||||||
3 | ||||||
4 | ||||||
5 |
Operations of FOMM_with_AWP and MRAA_with_AWP are the same.
Prepare your dataset. Recommend CASME2, SAMM, SMIC-HS
Divide into your_dataset/train
and your_dataset/test
Create or modify yaml
format file your_dataset_train.yaml
in ./config
Train
python run.py --config config/your_dataset_train.yaml
Log, parameters and checkpoints would be saved in ./log
Test
Create or modify csv
format file your_dataset_test.csv
in ./data
python run.py --config config/my_dataset_test.yaml --mode animate --checkpoint path/to/checkpoint
Generated videos would be saved in path/to/checkpoint/animation