基于对抗博弈策略的无人机红外图像自适应融合算法优化设计

Optimized Design of an Adaptive UAV Infrared Image Fusion Algorithm Based on Adversarial Game Strategy

  • 摘要: 针对无人机(Unmanned Aerial Vehicle, UAV)在复杂环境下红外-可见光图像融合存在的实时性不足、目标特征易丢失及背景伪影难抑制等问题,本文提出一种基于对抗博弈策略的无人机红外图像自适应融合算法。模型以轻量级RFN-Nest为骨干,构建生成器-判别器对抗框架,生成器利用通道-空间双重注意力实时加权红外热目标与可见光纹理;判别器基于真实融合图像监督全局真实性;并用红外统计掩码加权损失强化弱小热目标,实现20 M参数下的高精度实时融合,适合边缘设备部署。实验在TNO公开数据集上与次优方法相比,熵(Entropy, EN)提高1.8%,标准差(Standard Deviation, SD)提高5.1%,互信息(Mutual Information, MI)提高5.0%;同时将伪影评估因子(non-artifact blur factor, Nabf)降低9.4%,多尺度结构相似性(Multi-Scale Structural Similarity, MS-SSIM)提升0.7%,验证了算法的显著优势。

     

    Abstract: To address the challenges of insufficient real-time performance, loss of target features, and difficulty in suppressing background artifacts during unmanned aerial vehicle (UAV) infrared-visible image fusion in complex environments, an optimized adaptive fusion algorithm based on adversarial game strategies was proposed in this study. Built on the lightweight RFN-Nest backbone, the method established a generator–discriminator adversarial framework. The generator employed channel–spatial dual attention to dynamically weight infrared thermal targets and visible textures in real time, whereas the discriminator supervised global realism with real fused images. Furthermore, an infrared-statistics-based mask weighted loss was introduced to enhance weak thermal targets, achieving high-precision real-time fusion within 20 M parameters and enabling deployment on edge devices. On the public TNO dataset, compared with the second-best method, ROIMaskCNN, the proposed approach increased entropy (EN) by 1.8%, standard deviation (SD) by 5.1%, and mutual information (MI) by 5.0%; additionally, it reduced the non-artifact blur factor (Nabf) by 9.4% and improved multi-scale structural similarity (MS-SSIM) by 0.7%, demonstrating clear superiority.

     

/

返回文章
返回