​​​​​​​​​​​​​​​​​​摘要:U2Net 是一个优秀的显著性目标检测算法,由 Qin Xuebin 等人发表在 Pattern Recognition 2020 期刊[Arxiv]。U2Net 名称的来源在于其网络结构由两层嵌套的 Unet 结构,可以在不需要预训练骨干网络的情况下从零开始训练,拥有优异的表现。

本文分享自华为云社区《ModelArtsNotebook快速开源项目实战 — U2Net》,作者:shpity 。

一、U2Net 介绍

U2Net 是一个优秀的显著性目标检测算法,由 Qin Xuebin 等人发表在 Pattern Recognition 2020 期刊[Arxiv]。U2Net 名称的来源在于其网络结构由两层嵌套的 Unet 结构,可以在不需要预训练骨干网络的情况下从零开始训练,拥有优异的表现。其网络结构如图 1 所示。

U2Net基于ModelArts Notbook的仿真实验_U2Net

图 1. U2Net 的主体框架是一个类似于 U-Net 的编解码结构,但是每一个 block 替换为新提出的残差 U-block 模块

更多文章请关注《万象专栏》