tthinking/datfuse

[IEEE TCSVT 2023] Official implementation of DATFuse: Infrared and Visible Image Fusion via Dual Attention Transformer

DATFuse: Infrared and Visible Image Fusion via Dual Attention Transformer (IEEE TCSVT 2023)

This is the official implementation of the DATFuse model proposed in the paper (DATFuse: Infrared and Visible Image Fusion via Dual Attention Transformer) with Pytorch.

Comparison with SOTA methods

Fusion results on TNO dataset

Fusion results on RoadScene dataset

Ablation study on network structure

Ablation study on the number of TRMs

Ablation study on the second DARM

Impact of weight parameters in the loss function

Impact of weight parameter α on fusion performance with λ and γ fixed as 100 and 10, respectively.

Impact of weight parameter λ on fusion performance with α and γ fixed as 1 and 10, respectively.

Impact of weight parameter γ on fusion performance with α and λ fixed as 1 and 100, respectively.

Computational efficiency comparisons

Average running time for generating a fused image (Unit: seconds)

Method	TNO Dataset	RoadScene Dataset
MDLatLRR	26.0727	11.7310
AUIF	0.1119	0.0726
DenseFuse	0.5663	0.3190
FusionGAN	2.6796	1.1442
GANMcC	5.6752	2.3813
RFN_Nest	2.3096	0.9423
CSF	10.3311	5.5395
MFEIF	0.0793	0.0494
PPTFusion	1.4150	0.8656
SwinFuse	3.2687	1.6478
DATFuse	0.0257	0.0141

Cite the paper

If this work is helpful to you, please cite it as:

@ARTICLE{Tang_2023_DATFuse,
  author={Tang, Wei and He, Fazhi and Liu, Yu and Duan, Yansong and Si, Tongzhen},
  journal={IEEE Transactions on Circuits and Systems for Video Technology}, 
  title={DATFuse: Infrared and Visible Image Fusion via Dual Attention Transformer}, 
  year={2023},
  volume={33},
  number={7},
  pages={3159-3172},
  doi={10.1109/TCSVT.2023.3234340}}

If you have any questions, feel free to contact me ([email protected]).

datfuse's Issues

训练集

请问可以提供训练集的图片吗

好像缺少一个文件

FileNotFoundError: [Errno 2] No such file or directory: 'models/.../args.txt'

Hello author!
Thank you for releasing the code. I am trying to reproduce the experimental results mentioned in your paper, but when I was training , I found that the FileNotFoundError: [Errno 2] No such file or directory: 'models/.../args.txt', I would like to ask you what caused this？

问题请教GetDataset

打扰博主，请教一下问题。

输入 GetDataset 中的路径是'...'，对于里面的
def getitem(self, index):
ir = '...'
vi = '...'
请问这儿的 ir 和 vi 应该如何获取呢？填什么路径哇。对于GetDataset在 Train 中只分别传入了training_dir_ir和training_dir_vi 如何在GetDataset类中获取 ir 和 vi 图像路径呢？

唐博，请问下您论文中评价指标代码有吗？

normalized mutual information QM I [74],
Tsallis entropy QT E [75], nonlinear correlation information
entropy QNCIE [76], phase congruency based image feature
metric QP [77], structural similarity-based metric QY [78],
and Chen-Varshney metric QCV

训练集路径设置

train.py出现了这几个路径，请问ir='...'，vi='...'，training_dir_ir = "..."，training_dir_vi = "..."都怎么设置，后两个应该是红外和可见光图片的文件夹路径，前两个是什么啊

Dataset folder

The details of the training dataset folder，thanks

您好，请问是否可以开源一下TCCFuse的代码呢？

您好，请问是否可以开源一下TCCFuse的代码呢？谢谢.

好像缺少个文件