基于注意力机制的航拍图像实时语义分割方法
作者:袁旭亮; 王娟; 武明虎; 郭力权; 刘子杉; 陈关海
更新时间:
目前,语义分割网络存在推理速度慢、轮廓信息缺乏、语义信息不足等问题,不适合航空图像的语义分割。本文提出了一种基于交叉注意混合机制和金字塔注意机制的航空图像语义分割解码网络。首先,MobileNetV2被用作主干网络,以提高实时推理的速度。其次,提出了一种交叉注意的混合机制来解决轮廓信息缺失的问题。第三,提出了一种金字塔注意机制,以消除卷积神经网络无法捕捉远程语义信息的限制。最后,实验结果表明,该网络在GTX的一张3090卡片上获得73.4%的平均分辨率和85.4%的像素精度,在DL遥感Dataset数据集中的分辨率为2562563,推理速度为196.9帧/秒。