公开(公告)号：CN118429822A

公开(公告)日：2024-08-02

申请号：CN202410482046.5

申请日：2024-04-22

Applicant: 核工业北京地质研究院

Inventor： 高岩 ,  张怡婷 ,  陆冬华 ,  赵英俊

IPC: G06V20/13 ,  G06V20/10 ,  G06V10/764 ,  G06V10/80 ,  G06V10/774 ,  G06V10/82 ,  G06V10/25

Abstract: 本发明属于遥感图像目标检测领域，具体涉及一种基于改进YOLOv8的大范围遥感影像核电站目标检测方法，该方法包括：步骤S1，通过已知核电站坐标，使用影像下载模块从网络获取样本区域的遥感影像数据；步骤S2，在步骤S1获取的遥感影像数据上标注核电站位置及其所属类别，获得真实框，建立核电站样本数据集；步骤S3，改进YOLOv8的损失函数，使用BiFPN特征融合网络对YOLOv8主干网络提取的特征进行融合，构建基于改进YOLOv8的目标检测模型；步骤S4，将待检测的目标区域输入影像下载模块，影像下载模块自动获取范围内遥感影像，将遥感影像输入至步骤S3训练好的基于改进YOLOv8的目标检测模型进行检测，输出检测结果；步骤S5，对步骤S4输出的检测结果根据类别，目标最小外接矩形的相对目标区域坐标及目标置信度进行去冗，得到最终检测结果，并保存为相关格式的文件。本发明方法能够有效提高模型的检测精度、降低获取数据的成本、提高数据的利用率。