This repo implements the Invariant Feature aware Missing Modality Imagination Network(IF-MMIN) for the following paper: "Exploiting modality-invariant feature for robust multimodal emotion recognition with missing modalities"
python 3.8.0
pytorch >= 1.0.0
First you should change the data folder path in data/config
and preprocess your data follwing the code in preprocess/
.
You can download the preprocessed feature to run the code.
For Training IF-MMIN on IEMOCAP:
First training a model fusion model with all audio, visual and lexical modality as the pretrained encoder.
bash scripts/CAP_utt_shared.sh AVL [num_of_expr] [GPU_index]
Then
bash scripts/CAP_IFMMIN.sh [num_of_expr] [GPU_index]
Note that you can run the code with default hyper-parameters defined in shell scripts, for changing these arguments, please refer to options/get_opt.py and the modify_commandline_options
method of each model you choose.
Baidu Yun Link IEMOCAP A V L modality Features 链接: https://pan.baidu.com/s/1WmuqNlvcs5XzLKfz5i4iqQ 提取码: gn6w