Image-Fusion-Transformer
Platform
Python 3.7
Pytorch >=1.0
Training Dataset
MS-COCO 2014 (T.-Y. Lin, M. Maire, S. Belongie, J. Hays, P. Perona, D. Ramanan, P. Dollar, and C. L. Zitnick. Microsoft coco: Common objects in context. In ECCV, 2014. 3-5.) is utilized to train our auto-encoder network.
KAIST (S. Hwang, J. Park, N. Kim, Y. Choi, I. So Kweon, Multispectral pedestrian detection: Benchmark dataset and baseline, in: Proceedings of the IEEE conference on computer vision and pattern recognition, 2015, pp. 1037–1045.) is utilized to train the RFN modules.
The testing datasets are included in "analysis_MatLab".
Training Command:
python train_fusionnet_axial.py
Testing Command:
python test_21pairs_axial.py
The Fusion results are included in "analysis_MatLab".
If you have any questions about the code, feel free to contact me at [email protected].