巢湖学院学报 ›› 2023, Vol. 25 ›› Issue (3): 69-78.doi: 10.12152/j.issn.1672-2868.2023.03.009
梁伟,李莹莹,张硕:安徽建筑大学 电子与信息工程学院
LIANG Wei,LI Ying-ying,ZHANG Shuo:School of Electronic and Information Engineering, Anhui Jianzhu University
摘要: 针对遥感图像目标检测中小目标物体漏检率高、检测精度低的问题,提出了一种检测精度更高的遥感图像小目标检测方法EfficientNet-YOLOv3。该方法基于YOLOv3算法,采用EfficientNet-B0网络替换原YOLOv3算法的骨干网络,能更有效地提取图像特征;增加预测分支以及优化先验框的大小和个数,提高对遥感图像小目标的检测效果;同时选择DIoU为损失函数,提高目标先验框回归的效率,改善漏检现象。DOTA遥感图像数据集上的实验结果表明,算法平均精度均值(mAP)为91.01%,比原YOLOv3平均精度均值(mAP)提高了11.82%,具有更高的检测精度。
中图分类号: