
基于通道注意力机制的RGB-D图像的语义分割网络
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本研究提出了一种结合通道注意力机制的RGB-D图像语义分割网络,旨在提升深度和彩色信息融合效果,增强小目标识别精度。
针对RGB-D图像的语义分割问题,本段落提出了一种结合通道注意力机制的RefineNet网络。考虑到网络特征图中各个通道的重要性不同,将通道注意力机制分别引入基本RefineNet的编码器和解码器模块,以增强对重要特征的学习和关注;同时,使用focal loss函数替代传统的交叉熵损失函数来处理多类语义分割任务中的类别数量不平衡及难分样本问题。实验结果表明,在SUNRGBD和NYUv2数据集上,本段落网络在保持相近的参数量和计算量的同时,显著提高了分割精度,其mIOU分别达到45.7%和49.4%,优于最新的主流语义分割网络如Depth-aware、RDFNet 和Refinenet。
全部评论 (0)
还没有任何评论哟~


