基于DeepLab V3模型的图像语义分割速度优化研究

作者：司海飞; 史震^*; 胡兴柳; 杨春萍

来源：计算机工程与应用, 2020, 56(24): 137-143.

• 图像语义分割
• 移动端
• DeepLab V3模型
• 轻量化
• 卷积神经网络

为了解决移动端视觉感知模块的内存资源和硬件条件不适应卷积神经网络的快速应用等问题。以经典的DeepLab V3模型为基础,在保证分割精度的前提下,以优化卷积神经网络模型的运行速度为目标,提出一种新的轻量化网络结构。所设计的网络结构将原有非线性激活函数替换成新的Swish激活函数进行精度补偿,采用改进后的轻量化MobileNet V2结构替代DeepLab V3原有的特征提取器。实验结果表明,改进的DeepLab V3网络模型和目前精度最高的DeepLab V3+算法相比,其在维持一定精度的前提下,参数量和计算复杂度大大减小,运行速度明显提升,模型内存占用率下降了近96%,综合性能更强,更适合对分割性能要求较高的快速分割网络。

• 单位
  金陵科技学院; 哈尔滨工程大学