一种基于遥感图像的水利设施定向检测与标注方法

本发明属于遥感图像识别技术领域，具体提供一种基于遥感图像的水利设施定向检测与标注方法，包括：对水利设施的可见光遥感图像的尺寸进行调整；使用预先训练好的深度卷积神经网络作为模型的骨干网络，对所输入的遥感图像中水利设施视觉特征进行提取，将视觉特征从骨干网络传递到特征金字塔网络中；在深度卷积神经网络的末端设计神经网络，对每个水利设施类别的前景和背景进行区分，并对标注矩形的中心、长度、宽度和旋转角度进行回归；将将网络预测到的所有候选的定向框，通过基于定向框的非极大值抑制来进行合并，输出算法最终的定向框检测结果。本发明有助于可见光宽幅遥感图像中的角度分布任意的遥感目标的精细边界提取，可提升如水坝、观测站等水利设施的检测效率与标注精度。