UIU-Net：用于红外小目标检测的U-Net中的U-Net

佑福

一句话总结

U-Net in U-Net：一种用于红外小目标检测的新网络，其将微小的U-Net嵌入到更大的U-Net骨干网中，从而实现物体的多层次和多尺度表示学习，性能表现出色，代码刚刚开源！
点击关注@CVer计算机视觉，第一时间看到最优质、最前沿的CV、AI工作~
点击进入—>目标检测微信技术交流群
UIU-Net

UIU-Net: U-Net in U-Net for Infrared Small Object Detection


单位：北京邮电, 中科院
代码：https://github.com/danfenghong/IEEE_TIP_UIU-Net
论文（已收录TIP 2022）：https://arxiv.org/abs/2212.00968
基于学习的红外小目标检测方法目前严重依赖于分类骨干网。随着网络深度的增加，这往往会导致微小物体的丢失和特征可区分性限制。此外，红外图像中的小物体经常出现明亮和黑暗，对获得精确的物体对比度信息提出了严峻的要求。


为此，我们在本文中提出了一个简单有效的“U-Net in U-Net”框架，简称UIU-Net，用于检测红外图像中的小物体。


顾名思义，UIU-Net将微小的U-Net嵌入到更大的U-Net骨干网中，从而实现物体的多层次和多尺度表示学习。而且，UIU-Net可以从头开始训练，学习到的特征可以有效地增强全局和局部对比信息，


更具体地说，UIU-Net模型分为两个模块：分辨率-维护深度监督（RM-DS）模块和交互-交叉注意力（IC-A）模块。
RM-DS将残差U-块集成到深度监督网络中，在学习全局上下文信息的同时生成深度多尺度分辨率-维护特征


进一步地，IC-A对low-level细节和high-level语义特征之间的局部上下文信息进行编码。


实验结果

在两个红外单帧图像数据集，即SIRST和合成数据集上进行的广泛实验表明，与几种最先进的红外小目标检测方法相比，所提出的UIU-Net的有效性和优越性。








所提出的UIU-Net还为视频序列红外小目标数据集，例如ATR ground/air视频序列数据集提供了强大的泛化性能。


点击进入—>目标检测微信技术交流群
CVer-目标检测交流群

建了CVer-目标检测交流群！想要进目标检测学习交流群的同学，可以直接加微信号：CVer222。加的时候备注一下：目标检测+学校/公司+昵称+知乎，即可。然后就可以拉你进群了。
推荐阅读

扩散模型再下一城！DiffPose：使用扩散模型的多假设人体姿态估计
CCF论文列表(2022拟定)大更新！MICCAI空降B类！PRCV空降C类！ICLR继续陪跑...
CVer计算机视觉：SuperYOLO：用于多模态遥感图像中的超分辨率辅助目标检测
武汉大学提出ElegantSeg：用于超大图像语义分割的端到端整体学习
首个！SinFusion：在单个图像或视频上训练扩散模型
把扩散模型应用到3D医学图像重建！DiffusionMBIR：使用预训练2D扩散模型求解3D逆问题
DiffusionDet：第一个用于目标检测的扩散模型
CVer计算机视觉：刷新图像分割新记录！OneFormer：规则通用图像分割的Transformer
65.4 AP刷新COCO目标检测新记录！InternImage：探索具有可变形卷积的大规模视觉基础模型
RepGhost：通过重参数化实现硬件高效的Ghost模块
64.5 AP！Group DETR v2：编-解码器预训练的强大目标检测器
NeurIPS 2022 | 香港理工提出OGC：首个无监督3D点云物体实例分割算法
Sea和北大提出新优化器Adan：深度模型都能用！训练ViT和MAE减少一半计算量！
MedSegDiff：基于扩散概率模型的医学图像分割
NeurIPS 2022 | 清华&南开提出SegNeXt：重新思考语义分割的卷积注意力设计
YOLOv4团队打造YOLOv7！最先进的实时目标检测网络来了！
COCO上高达62.4 AP！GLIPv2来了！统一定位和视觉-语言理解
FAIR提出ConvNeXt：2020 年代的卷积网络
清华提出：最新的计算机视觉注意力机制(Attention)综述！
为何Transformer在计算机视觉中如此受欢迎？
Transformer一脚踹进医学图像分割！看5篇MICCAI 2021有感

莘蒂蕊拉

这不是前几年做显著性比较经典的U2NET吗，结构一样的[飙泪笑][飙泪笑]

袁头

u2net？

西门无雪

U2net换个皮罢了 现在的文章质量越来越低了

七分醉意

真是换皮王...

盘古九哥

6666

快乐天天

只能说明U2Net还是强阿

少年风

[尴尬][尴尬][尴尬]

谷俊山水

红外小目标检测，数据集好像一直都没有官方的吧。不像做目标检测有COCO之类的。我看了几篇红外小目标检测的文章，数据集都不同。这个怎么比较？

拐得很

好帖必须得顶起