摘要
针对现有语义分割网络模型难以在参数量、推理速度和精确度中取得平衡的问题,设计一种多尺度特征信息融合的轻量级网络模型MIFNet(Multiscale-Information-Fusion Network)。MIFNet采用编码-解码结构,在编码部分利用分离策略和非对称卷积设计了轻量型特征提取瓶颈结构,且引入空间注意力机制与Laplace边缘检测算子组成边缘-空间融合模块,将空间信息和边缘信息进行融合得到丰富的特征信息。在解码部分引入通道注意力机制恢复特征图尺寸和细节信息完成语义分割。在Cityscapes和CamVid测试集上,MIFNet仅以0.82M的参数量分别取得了73.1%和67.7%mIoU,同时在单个GTX 1080Ti GPU下分别获得73.68 fps和85.16 fps的推理速度,表明该方法在参数量、推理速度和精确度三个指标上得到较好平衡,实现了轻量、快速、精准的语义分割。
- 单位