首页|基于区域掩码对比蒸馏的遥感图像目标检测

基于区域掩码对比蒸馏的遥感图像目标检测

扫码查看
为解决遥感图像中存在背景干扰以及目标密集分布的问题,采用了一种以区域掩码对比蒸馏为基础的目标检测方法,旨在提高遥感图像目标检测性能.首先,通过对目标特征区域进行掩码操作,以区分前景和背景并捕捉目标的细节纹理,生成细致的特征掩码.其次,结合对比蒸馏算法,通过对教师网络和学生网络的区域掩码进行对比学习,使学生网络更加充分地吸收教师网络在目标特征纹理检测方面的知识.同时,在检测阶段引入了一种旋转定位损失算法,该算法通过量化真实边界框和预测边界框之间的格点向量差来进行损失估计,从而减小了预测边界框与真实边界框之间的旋转损失.结果表明,改进算法的均值平均精度在DOTA和HRSC2016数据集上分别较传统算法提高了 3.57%和5.22%.
Object detection in remote sensing images based on region mask contrastive distillation
To address the challenges of background interference and dense target distribution in remote sensing images,a target detection method based on region mask contrastive distillation was implemented.The method aims to improve target detection performance in remote sensing images.Initially,detailed feature masks were created by applying masking operations to specific target regions,distinguishing foreground from background and capturing intricate target textures.Subsequently,the contrastive distillation algorithm was utilized,facilita-ting a comparison-based learning approach between the region masks of teacher and student networks.This ap-proach allowed the student network to comprehensively absorb the teacher network's knowledge related to de-tecting target feature textures.Concurrently,a rotational positioning loss function was introduced during the detection phase.This algorithm estimated loss by measuring grid vector differences between ground truth and predicted bounding boxes,thus reducing rotational disparities between predicted and ground truth bounding bo-xes.The results demonstrate that the mean average precision of the improved algorithm is 3.57%and 5.22%higher than that of the traditional algorithm on the DOTA and HRSC2016 datasets,respectively.

remote sensing imageobject detectionregion mask contrastive distillationrotational positio-ning loss

周杰、周子龙、罗岩、刘瑞、赵满艳

展开 >

南京信息工程大学电子与信息工程学院,南京 210044

遥感图像 目标检测 区域掩码对比蒸馏 旋转定位损失函数

国家自然科学基金面上资助项目国家自然科学基金面上资助项目国家自然科学基金面上资助项目

619711676210127462101275

2024

东南大学学报(自然科学版)
东南大学

东南大学学报(自然科学版)

CSTPCD北大核心
影响因子:0.989
ISSN:1001-0505
年,卷(期):2024.54(3)