一种视觉提示引导的遥感影像相似目标检测方法及装置

本发明公开了一种视觉提示引导的遥感影像相似目标检测方法和装置。方法包括：遥感数据选取和预处理，对图像和视觉提示进行数据增强，获得预处理数据集；构建目标检测模型；利用CNN主干网络对预处理图像进行特征提取输入到Transformer的编码器中；利用CNN主干网络对视觉提示进行特征提取后与查询对象相加，输入到Transformer的解码器中；预处理图像和视觉提示的处理数据在解码器中融合，再经FNN输出边界框和标签，得到预测结果；利用损失函数和预测结果，对网络进行训练，获得最终目标检测模型。该方法通过将多分类任务转变为二分类任务(匹配或者不匹配)，以及对采用的遥感数据集进行处理，并引入视觉提示Visual Prompt，从而将DETR模型扩展到相似目标检测任务，并提高模型性能。