.png){kind=icon}
返回
ISTD-diff: Infrared Small Target Detection via Conditional Diffusion Models
.png){kind=icon}
PRE
AI
DOI：10.1109/LGRS.2024.3401838.png)
.png){kind=icon}
.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Infrared small-target detection (IRSTD), which is to extract tiny and dim targets that are hidden in noisy and messy backgrounds, is a challenging task in computer vision. Inspired by the recently powerful deep generative models, we formulate the IRSTD as a generative task and design a conditional denoising (DE) model termed ISTD-diff to iteratively generate the target mask from the noisy one. In addition, ISTD-diff employs a two-pathway architecture, consisting of a conditional prior (CP) stream for encoding the input infrared image prior and a DE stream for cleaning up the noisy masks. Both streams are equipped with several cascaded innovative channel-dimension transformer (CDT) layers, which capture the global correlations efficiently and reduce computational demands effectively. Moreover, to strengthen the DE learning process, we proposed a simple, but powerful method named attention injection module (AIM), which provides detailed control over the DE stream. Extensive experiments finely demonstrate the superior performance of our ISTD-diff beyond the current representative segmentation-based state-of-the-art (SOTA) algorithms.
Keyword:
Streaming media
Task analysis
Noise reduction
Transformers
Correlation
Object detection
Noise measurement
Diffusion model
dual-dimension transformer
feature injection
infrared small-object detection
期刊
IF:
16.4
论文数:
1.0W
被引数:
5.1K
机构
引用论文
SRDiff: Single image super-resolution with diffusion probabilistic modelsSRDiff: 具有扩散概率模型的单幅图像超分辨率.png)
NEUROCOMPUTING
OAAI
NEUROCOMPUTINGIF6.5
OAAI分享收藏Infrared Small and Dim Target Detection With Transformer Under Complex Backgrounds复杂背景下基于变压器的红外弱小目标检测
PREAI
分享收藏Infrared Dim and Small Target Detection via Multiple Subspace Learning and Spatial-Temporal Patch-Tensor Model基于多子空间学习和时空补丁张量模型的红外弱小目标检测
PREAI
分享收藏Infrared Small Target Detection Based on the Weighted Strengthened Local Contrast Measure基于加权增强局部对比度测度的红外小目标检测
PREAI
分享收藏分享收藏
查看更多内容