3 个月前

面向室内环境的高效多任务RGB-D场景分析

面向室内环境的高效多任务RGB-D场景分析

摘要

语义场景理解对于在多种环境中运行的移动智能体至关重要。尽管语义分割已提供大量信息,但个体物体的细节以及整体场景的完整理解仍存在缺失,而这对于许多现实世界的应用而言却是必需的。然而,在移动平台受限的计算能力和电池资源条件下,若分别求解多个任务,不仅成本高昂,且难以实现实时处理。本文提出一种高效的RGB-D场景分析多任务框架(EMSANet),能够实时同步完成语义分割与实例分割(即全景分割)、实例方向估计以及场景分类等多项任务。我们证明,通过单一神经网络即可在移动平台上实现实时处理,且不牺牲性能;相反,各任务之间还能相互促进、协同提升。为评估所提出的多任务方法,我们扩展了常用RGB-D室内数据集NYUv2和SUNRGB-D的标注,新增实例分割与方向估计的标注信息。据我们所知,本工作是首个在NYUv2和SUNRGB-D数据集上实现如此全面多任务设置的室内场景分析研究,为该领域提供了新的基准。

代码仓库

tui-nicr/emsanet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
panoptic-segmentation-on-nyu-depth-v2EMSANet
PQ: 47.38
panoptic-segmentation-on-sun-rgbdEMSANet
PQ: 52.84
semantic-segmentation-on-nyu-depth-v2EMSANet (2x ResNet-34 NBt1D, finetuned)
Mean IoU: 53.34%
semantic-segmentation-on-sun-rgbdDPLNet
Mean IoU: 48.47%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
面向室内环境的高效多任务RGB-D场景分析 | 论文 | HyperAI超神经