This repository is Pytorch code for our proposed joint Modality and Pattern Alignment Network (MPANet)
PyTorch == 1.2.0
ignite == 0.2.1
torchvision == 0.4.0
apex == 0.1
git clone https://github.com/MPANet/MPANet.git
The path to datasets can be modified in the following file:
./configs/default/dataset.py
To train the model, you can use following command:
SYSU-MM01:
python train.py --cfg ./configs/SYSU.yml
RegDB:
python train.py --cfg ./configs/RegDB.yml
The checkpoint can be found here: https://pan.baidu.com/s/1TnjtfMFPnm5TEprgAhqz9A
Code: rfti