基于深度学习的无人机地物图像分割方法

针对无人机地物图像的传统分割方法工程量大，效率低下，深度学习的无人机地物图像分割算法在复杂场景下精度不高和数据集的类别不均衡（长尾数据）等问题，提出一种基于深度学习的高分辨率无人机地物图像分割方法，用于提高不同地貌区域的分割精度。在语义分割模型DeepLabv3的基础上进行改进，将原始主干网络ResNet101替换为ResNet152并添加预训练模型，调整扩张卷积空间金字塔池化模块的扩张率，采用类别平衡损失函数来解决长尾数据问题。在采集的无人机地物图像数据集上进行训练并通过测试集的分割效果证明模型改进方法的有效性。根据实验模型分割效果表明，改进后的方法在测试集上平均交并比达到70.8%，相比原始模型提升了27.2%，能够得到效果更好的分割结果。