首页 | 本学科首页   官方微博 | 高级检索  
     

基于改进YOLOv5s的遥感目标检测
引用本文:王悦炜,焦良葆,高阳. 基于改进YOLOv5s的遥感目标检测[J]. 计算机测量与控制, 2023, 31(8): 70-76
作者姓名:王悦炜  焦良葆  高阳
作者单位:南京工程学院 人工智能产业技术研究院,,,
基金项目:江苏省自然科学基金资助项目(BK20201042);江苏省政策引导类计划项目(SZ-SQ2020007)
摘    要:优化遥感目标检测对于军事民生方面都有着重要意义。由于遥感数据中成像模糊、目标较小、检测对象数量多,导致检测精度不高,现提出一种新网络:新网络在YOLOv5s(You Only Look Once v5s)原网络的基础上使用Mish激活函数代替SiLU(Sigmoid Weighted Liner Unit)激活函数;为解决遥感图像中小目标的问题,采用了对小目标、低分辨率友好的SPD-Conv(Space-to-depth-Conv)模块;考虑到使用耦合检测头会存在回归、分类两个任务之间的冲突问题,采用YOLOX (You Only Look Once X)中解耦的检测头,提高了模型查准率。实验结果表明,相比于原始YOLOv5s,新网络在mAP(mean Average Precision)平均精度均值方面提升了7%,查全率(recall)提升了10.9%,检测速度FPS提升了16.95%。改进后的网络模型相对于原始模型具有明显优越性,识别效果提升明显。

关 键 词:深度学习  遥感  激活函数  小目标   解耦检测头
收稿时间:2023-02-01
修稿时间:2023-03-04

Remote sensing target detection based on improved YOLOv5s
Abstract:Optimizing remote sensing target detection is of great significance to military and people''s livelihood. Due to the blurred image, small target and large number of detected objects in the remote sensing data, the detection accuracy is not high. A new network is proposed: the new network uses the Mish activation function to replace the SiLU (Sigmaid Weighted Liner Unit) activation function on the basis of the original YOLOv5s (You Only Look Once v5s) network; In order to solve the problem of small targets in remote sensing images, SPD-Conv (Space-to-depth-Conv) module which is friendly to small targets and low resolution is adopted; Considering the conflict between regression and classification tasks when using coupling detector heads, the decoupled detector heads in YOLOX (You Only Look Once X) are used to improve the model precision. The experimental results show that compared with the original YOLOv5s, the new network has improved the average accuracy of mAP (mean average precision) by 7%, the recall rate by 10.9% and the detection speed by 16.95%. The improved network model has obvious advantages over the original model, and the recognition effect is significantly improved.
Keywords:deep learning   remote sensing   activation function   small goals    decoupling detector
点击此处可从《计算机测量与控制》浏览原始摘要信息
点击此处可从《计算机测量与控制》下载免费的PDF全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号