论文阅读笔记——《室内服务机器人的实时场景分割算法》

轻量级深度学习模型在图像分割中的应用
该文提出了一种基于残差瓶颈结构的轻量级网络,结合深度可分离卷积、膨胀卷积和通道注意力机制,实现高精度的图像分割。在NYUDv2和CamVid数据集上,模型的MIoU分别达到72.7%和59.9%,且计算力仅为4.2GFLOPs,参数量为8.3Mb。网络采用编码器-解码器架构,通过多尺度特征融合和残差连接来保持空间信息并提高分割精度。

一、主要工作

  • 通过深度可分离卷积膨胀卷积通道注意力机制设计轻量级的高准确度特征提取模块。
  • 融合浅层特征与深层语义特征获得更丰富的图像特征。
  • 在NYUDv2和CamVid数据集上的MIoU分别达到72.7%和59.9%,模型的计算力为4.2GFLOPs,参数量为8.3Mb。

二、残差瓶颈结构

  为了减小参数量同时达到高分割精度,设计了一个残差瓶颈结构作为网络的基础模块,包括2个调整特征通道的1x1卷积、1个深度可分离膨胀卷积、1个通道注意力模块和一条残差连接。

在这里插入图片描述

图2-1 残差瓶颈结构

  设计思路如下:

  • 降维-升维的瓶颈结构。先使用1×1的卷积对通道数进行压缩,压缩后使用3×3深度可分离卷积提取特征,再使用1×1卷积进行特征升维。为增大感受野,提取不同尺度的特征,3×3深度可分离卷积带有膨胀系数。
  • 通道注意力机制模块。在升维卷积操作前加入SE-Net网络的通道注意力机制模块,计算过程如图2-2所示,首先对特征通道进行全局平均池化操作,然后连接全连接层学习每个通道的权重值,并利用Sigmoid函数归一化到[0,1],最后对特征图进行通道加权,从而筛选特征的每个通道的信息,以此进一步提高分割精度。

在这里插入图片描述

图2-2 通道注意力模块计算过程
评论 2
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

晓shuo

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值