电光与控制, 2023, 30 (6): 60, 网络出版: 2023-11-29  

基于改进YOLO模型的遥感小目标检测

Remote Sensing Small Target Detection Based on Improved YOLO Model
王凯 1,2王伟 1,2蒋志伟 1,2
作者单位
1 河北工业大学电子信息工程学院, 天津 300000
2 天津市电子材料与器件重点实验室, 天津 300000
摘要
针对遥感图像中小目标检测精度低以及漏检现象严重的问题, 提出一种基于YOLOv4改进的遥感小目标检测算法。该算法首先改进特征提取网络, 删除深层次特征层, 减少语义丢失现象;其次将轻量级注意力机制与RFB-S结构融合, 拓展感受野, 并加强网络对重要信息的关注程度, 从而提升检测精度;最后使用Focal Loss函数解决正负样本不均衡问题, 抑制背景目标, 进一步增强检测效果。在RSOD数据集上的实验结果表明, 改进后算法检测平均精度为96.5%, 召回率达到87.2%, 检测效果明显提升, 有效改善了小目标漏检现象, 对遥感图像小目标检测具有重要意义。
Abstract
A remote sensing small target detection algorithm based on the improved YOLOv4 is proposed to address the problems of low detection accuracy and serious missing detection of small targets in remote sensing images.Firstly, the feature extraction network is improved by removing the deep feature layer to reduce semantic loss.Secondly, the lightweight attention mechanism is fused with the RFB-S structure to expand the perceptual field and enhance attention to important information, thus improving the detection precision.Finally, the Focal Loss is used to avoid the imbalance between positive and negative samples and suppress the background targets to further enhance the detection effect.The experimental results on the RSOD dataset show that the improved algorithm has an average detection precision of 96.5% and a recall rate of 87.2%, which significantly improves the detection effect and effectively avoids the phenomenon of small target miss detection, and is of great significance to small target detection in remote sensing images.
参考文献

[1] 翟旭升,杨仕美,段朋振,等.航空发动机非线性分布式控制系统故障检测[J].电光与控制,2020,27(7):106-110.

[2] 姜文志,李炳臻,顾佼佼,等.基于改进 YOLO V3 的舰船目标检测算法[J].电光与控制,2021,28(6):52-56,67.

[3] 何迪.基于深度学习的肺部X光图像分类与目标检测[D].西安:西安电子科技大学,2021.

[4] GIRSHICK R,DONAHUE J,DARRELL T,et al.Rich feature hierarchies for accurate object detection and semantic segmentation[C]//IEEE Conference on Computer Vision and Pattern Recognition.Columbus,OH:IEEE,2014:580-587.

[5] REDMON J,DIVVALA S,GIRSHICK R,et al.You only look once:unified,real-time object detection[C]//IEEE Confer-ence on Computer Vision and Pattern Recognition.Las Vegas, NV:IEEE,2016:779-788.

[6] LIU W,ANGUELOV D,ERHAN D,et al.SSD:single shot multibox detector[C]//European Conference on Computer Vision.Cham:Springer,2016:21-37.

[7] LIN T Y,DOLLAR P,GIRSHICK R,et al.Feature pyramid networks for object detection[C]//IEEE Conference on Computer Vision and Pattern Recognition.Honolulu, HI:IEEE,2017:936-944.

[8] 戴伟聪,金龙旭,李国宁,等.遥感图像中飞机的改进 YOLOv3实时检测算法[J].光电工程,2018,45(12):84-92.

[9] KISANTAL M,WOJNA Z,MURAWSKI J,et al.Augmentation for small object detection[EB/OL].(2019-02-19)[2022-03-26].https://arxiv.org/abs/1902.07296.

[10] 刘峰,郭猛,王向军.基于跨尺度融合的卷积神经网络小目标检测[J].激光与光电子学进展,2021,58(6):213-221.

[11] FU C Y,LIU W,RANGA A,et al.DSSD:deconvolutional single shot detector[EB/OL].(2017-01-23)[2022-03-26].https: //arxiv.org/abs/1701.06659.

[12] 王俊强,李建胜,周学文,等.改进的SSD算法及其对遥感影像小目标检测性能的分析[J].光学学报,2019, 39(6):373-382.

[13] 徐逸之,姚晓婧,李祥,等.基于全卷积网络的高分辨遥感影像目标检测[J].测绘通报,2018(1):77-82.

[14] 张裕,杨海涛,刘翔宇.基于多尺度特征稠密连接的遥感图像目标检测方法[J].中国电子科学研究院学报, 2019, 14(5):530-536.

[15] BOCHKOVSKIY A,WANG C Y,LIAO H Y M.YOLOv4:optimal speed and accuracy of object detection[EB/OL].(2020-04-24)[2022-03-26].https://arxiv.org/abs/2004.10934.

[16] LIU S,HUANG D,WANG Y.Receptive field block net for accurate and fast object detection[C]//European Conference on Computer Vision (ECCV).Munich:Sprin-ger,2018:404-419.

[17] LIN T Y,GOYAL P,GIRSHICK R,et al.Focal loss for dense object detection[C]//IEEE International Conference on Computer Vision.Venice:IEEE,2017:2999-3007.

[18] HE K,ZHANG X,REN S,et al.Spatial pyramid pooling in deep convolutional networks for visual recognition[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2015,37(9):1904-1916.

[19] WEN L,LI X,GAO L.A transfer convolutional neural network for fault diagnosis based on ResNet-50[J].Neural Computing and Applications,2020,32(10):6111-6124.

[20] DE BOER P T,KROESE D P,MANNOR S,et al.A tutorial on the cross-entropy method[J].Annals of Operations Research,2005,134(1):19-67.

[21] LONG Y,GONG Y,XIAO Z,et al.Accurate object localization in remote sensing images based on convolutional neural networks[J].IEEE Transactions on Geoscience and Remote Sensing,2017,55(5):2486-2498.

王凯, 王伟, 蒋志伟. 基于改进YOLO模型的遥感小目标检测[J]. 电光与控制, 2023, 30(6): 60. WANG Kai, WANG Wei, JIANG Zhiwei. Remote Sensing Small Target Detection Based on Improved YOLO Model[J]. Electronics Optics & Control, 2023, 30(6): 60.

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!