一种尾矿库智能识别方法和装置
摘要:
本发明公开了一种尾矿库智能识别方法和装置。其中,该方法包括:对获取到的遥感图像进行预处理,以得到预处理后的遥感图像;对所述预处理后的遥感图像进行图像增强操作,以得到增强后的遥感图像;基于深度学习目标检测模型对所述增强后的遥感图像进行训练,以得到训练后的尾矿库识别模型;将待识别的目标尾矿库图像输入所述尾矿库识别模型,以得到所述尾矿库识别模型输出的目标尾矿库的识别结果。本发明通过将目标检测网络RetinaNet引入遥感影像尾矿库识别任务,通过深度残差网络框架ResNet101以及特征金字塔结构提取并融合图像中的浅层纹理信息及深层语义信息,通过焦点损失目标函数调节正负样本的损失权重,实现了遥感影像中尾矿库的精确定位。
0/0