摘要
针对遥感影像复杂背景和小目标检测困难的问题,提出了一种基于多感知融合的检测算法YOLO-GT。为了提升特征图中小目标的特征信息,设计了包含3种感知机制的检测头 (Adaptive Scale-Aware Dynamic Head,ASADH);同时,引入轻量级上采样算子CARAFE,解决语义信息丢失问题,提升特征金字塔网络性能;接着为进一步优化模型的训练速度和定位精度,采用了Wise-IOU作为损失函数。实验结果在DIOR数据集上显示,模型精度达90.4%,比原算法提高2.1%。这些改进有效提高了复杂背景下遥感影像小目标的检测性能。
-
单位四川轻化工大学