科研之友
微信
新浪微博
Facebook
分享链接
摘要
针对轻量语义分割算法应用于无人机高分辨率交通场景图像分割时存在边缘信息模糊、小目标特征提取准确性较差的问题，提出一种融合注意力机制与重影特征映射的轻量级语义分割算法。首先在BiSeNet V2算法语义分支8倍和16倍下采样过程嵌入混合注意力模块，重新分配深层特征图权重，增强局部关键特征提取能力；然后采用重影特征映射单元优化传统卷积层，进一步降低运算成本；最后使用动态阈值损失函数监督训练，调节高损失困难样本训练权重。利用UAVid数据集对改进后的算法进行训练并测试，发现算法平均交并比(mean intersection over union, mIoU)为52.7%,较改进前的模型提升7.8%,且当输入图像尺寸为1 280×736时推理速度达到81.6 FPS,满足实时分割要求。结果表明，该算法能较好适应复杂交通场景，有效改善边缘信息模糊和小目标分割准确性较差的问题。
- 单位
