室内服务机器人的实时场景分割算法

• 作者: 林杰 陈春梅 刘桂华 祝礼佳
• 来源期刊: 计算机工程
• 关键词: 轻量化网络 场景分割 深度可分离卷积 膨胀卷积 注意力机制
• 年，卷（期）: 2021,（7）
• 所属期刊栏目: 热点与综述|Research Hotspots and Reviews
• 页码范围: 21-29
• 页数: 9页
• 分类号: TP391
• 语种: 中文
• DOI: 10.19678/j.issn.1000-3428.0059577

摘要

室内场景下的实时场景分割是开发室内服务机器人的一项关键技术,目前关于语义分割的研究已经取得了重大进展,但是多数方法都倾向于设计复杂的网络结构或者高计算成本的模型来提高精度指标,而忽略了实际的部署成本.针对移动机器人算力成本有限的问题,设计一种轻量化的瓶颈结构,并以此为基本元素构建轻量化场景分割网络.该网络通过与特征提取网络级联获得更深层次的语义特征,并且融合浅层特征与深层语义特征获得更丰富的图像特征,其结合深度可分离卷积与多尺度膨胀卷积提取多尺度图像特征,减少了模型的参数量与计算量,同时利用通道注意力机制提升特征加权时的网络分割精度.以512像素×512像素的图像作为输入进行实验,结果表明,该算法在NYUDv2室内场景分割数据集和CamVid数据集上的MIoU分别达到72.7％和59.9％,模型计算力为4.2 GFLOPs,但参数量仅为8.3 Mb,在移动机器人NVIDIA Jetson XavierNX嵌入式平台帧率可达到42 frame/s,其实时性优于DeepLabV3+、PSPNet、SegNet和UNet算法.