一种基于SAM的端到端遥感目标检测方法、电子设备及存储介质

    公开(公告)号:CN119251700A

    公开(公告)日:2025-01-03

    申请号:CN202411394163.2

    申请日:2024-10-08

    Abstract: 一种基于SAM的端到端遥感目标检测方法、电子设备及存储介质,属于遥感图像目标检测技术领域。为了将大模型技术应用于遥感领域,本发明采集RGB可见光遥感图像,构建遥感数据集;构建特征提取的并行双分支骨干网络;构建特征融合模块,将得到的特征提取的特征数据输入到特征融合模块中,输出特征融合的特征数据,输入到提示生成器中,生成提示信息数据;将特征融合的特征数据和提示信息数据,输入到冻结后的SAM的掩码解码器,计算掩码的水平最大的长和垂直最大的宽,以掩码的中心坐标绘制检测框实现对目标的检测;构建损失函数,所述损失函数包括语义头的分类损失、定位头的回归损失和冻结的SAM掩码解码器生成掩码的分割损失构成。

Patent Agency Ranking