HyperAIHyperAI

Command Palette

Search for a command to run...

U2^22-Net:通过嵌套U结构深化显著性物体检测

Xuebin Qin, Zichen Zhang, Chenyang Huang, Masood Dehghan, Osmar R. Zaiane, Martin Jagersand

摘要

在本文中,我们设计了一种简单而强大的深度网络架构——U2^22-Net,用于显著目标检测(SOD)。U2^22-Net 的架构是一种两级嵌套的 U 形结构。该设计具有以下优势:(1) 由于我们在提出的残差 U 块(ReSidual U-blocks, RSU)中混合了不同大小的感受野,因此能够从不同尺度捕获更多的上下文信息;(2) 通过在这些 RSU 块中使用池化操作,可以在不显著增加计算成本的情况下增加整个架构的深度。这种架构使我们能够在不依赖图像分类任务中的骨干网络的情况下,从零开始训练一个深层网络。我们实例化了两种模型:U2^22-Net(176.3 MB,GTX 1080Ti GPU 上达到 30 FPS)和 U2^22-Net^{\dagger}(4.7 MB,40 FPS),以适应不同的应用场景。这两种模型在六个 SOD 数据集上均取得了具有竞争力的性能。代码已公开:https://github.com/NathanUA/U-2-Net


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供