3 个月前

SpatialFlow:连接全景分割的所有任务

SpatialFlow:连接全景分割的所有任务

摘要

目标定位是全景分割中的基础性任务,因为它与图像场景中的所有“事物”(things)和“东西”(stuff)密切相关。准确掌握图像中目标的位置信息,能够为分割提供重要线索,并帮助网络更深入地理解场景语义。如何在“事物”与“东西”的分割任务中有效融合目标位置信息,是一个关键挑战。本文提出一种空间信息流(spatial information flows)机制,以实现该目标。该信息流能够通过将边界框回归任务中获得的目标空间上下文传递至其他子任务,从而实现全景分割中各子任务之间的有效连接。更重要的是,我们设计了四个并行的子网络,以在不同子任务中更优地适配目标的空间信息。基于这些子网络与信息流结构,我们构建了一个具备位置感知能力且统一的全景分割框架,命名为 SpatialFlow。我们对框架中的每个组件进行了详尽的消融实验,并通过大量实验证明了 SpatialFlow 的有效性。此外,该方法在 MS-COCO 和 Cityscapes 两个主流全景分割基准上分别取得了 47.9 PQ 和 62.5 PQ 的领先性能。代码将开源,地址为:https://github.com/chensnathan/SpatialFlow。

代码仓库

chensnathan/SpatialFlow
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
panoptic-segmentation-on-coco-test-devSpatialFlow(ResNet-101-FPN)
PQ: 48.5
PQst: 37.9
PQth: 55.5

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
SpatialFlow:连接全景分割的所有任务 | 论文 | HyperAI超神经