4 个月前

RedNet:用于室内RGB-D语义分割的残差编码器-解码器网络

RedNet:用于室内RGB-D语义分割的残差编码器-解码器网络

摘要

室内语义分割一直是计算机视觉领域的难题。本文提出了一种名为RedNet的RGB-D残差编码器-解码器架构,用于室内RGB-D语义分割。在RedNet中,残差模块作为基本构建块应用于编码器和解码器,并使用跳跃连接(skip-connection)来传递编码器和解码器之间的空间特征。为了融合场景的深度信息,我们构建了一个融合结构,该结构分别对RGB图像和深度图像进行推理,并在多个层上融合它们的特征。为了高效地优化网络参数,我们提出了一种“金字塔监督”训练方案,该方案在解码器的不同层应用监督学习,以解决梯度消失的问题。实验结果表明,所提出的RedNet(基于ResNet-50)在SUN RGB-D基准数据集上达到了最先进的mIoU精度47.8%。

基准测试

基准方法指标
semantic-segmentation-on-nyu-depth-v2RedNet
Mean IoU: 47.2%
semantic-segmentation-on-sun-rgbdTokenFusion (Ti)
Mean IoU: 47.8%
semantic-segmentation-on-thud-robotic-datasetRedNet
mIoU: 76.92

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
RedNet:用于室内RGB-D语义分割的残差编码器-解码器网络 | 论文 | HyperAI超神经