提出统一框架UniCD,同时优化红外无人机图像去非均匀性和目标检测性能。
Detection-Friendly Nonuniformity Correction: A Union Framework for Infrared UAVTarget Detection

- 将非均匀性校正建模为参数估计问题,联合先验与数据生成利于检测的图像。
- 引入目标掩码监督的辅助损失,增强目标特征并抑制背景干扰。
- 设计检测引导的自监督损失,提升对不同非均匀程度的鲁棒性,适合实时监控场景。
采用热探测器获取的红外无人机图像常受温度相关低频非均匀性影响,显著降低图像对比度。在非均匀条件下检测无人机目标对无人机监视应用至关重要。现有方法通常将红外非均匀性校正(NUC)作为检测前的预处理步骤,导致性能不理想。在提升检测信息的同时平衡校正任务仍具挑战。本文提出一种检测友好的统一框架UniCD,以端到端方式同步解决红外NUC与无人机目标检测任务。首先,将NUC建模为少量参数估计问题,通过先验与数据联合驱动生成利于检测的图像;其次,在红外无人机目标检测网络主干中引入新的带目标掩码监督的辅助损失,强化目标特征并抑制背景;为更好平衡校正与检测,设计检测引导的自监督损失,减少两任务间特征差异,提升检测对不同非均匀水平的鲁棒性。此外,构建包含5万张红外图像的新基准IRBFD,涵盖多种非均匀类型、多尺度无人机目标和丰富背景,附带目标标注。在IRBFD上的大量实验表明,UniCD是鲁棒的统一框架,具备实时处理能力。数据集可于https://github.com/IVPLaboratory/UniCD 获取。
原文摘要 · Abstract (English)
Infrared unmanned aerial vehicle (UAV) images captured using thermal detectors are often affected by temperature dependent low-frequency nonuniformity, which significantly reduces the contrast of the images. Detecting UAV targets under nonuniform conditions is crucial in UAV surveillance applications. Existing methods typically treat infrared nonuniformity correction (NUC) as a preprocessing step for detection, which leads to suboptimal performance. Balancing the two tasks while enhancing detection beneficial information remains challenging. In this paper, we present a detection-friendly union framework, termed UniCD, that simultaneously addresses both infrared NUC and UAV target detection tasks in an end-to-end manner. We first model NUC as a small number of parameter estimation problem jointly driven by priors and data to generate detection-conducive images. Then, we incorporate a new auxiliary loss with target mask supervision into the backbone of the infrared UAV target detection network to strengthen target features while suppressing the background. To better balance correction and detection, we introduce a detection-guided self-supervised loss to reduce feature discrepancies between the two tasks, thereby enhancing detection robustness to varying nonuniformity levels. Additionally, we construct a new benchmark composed of 50,000 infrared images in various nonuniformity types, multi-scale UAV targets and rich backgrounds with target annotations, called IRBFD. Extensive experiments on IRBFD demonstrate that our UniCD is a robust union framework for NUC and UAV target detection while achieving real-time processing capabilities. Dataset can be available at https://github.com/IVPLaboratory/UniCD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。