| 针对低空微小型无人机的轻量型YOLOv5检测算法 |
| |
| 引用本文： | 魏峰,周建平,谭翔,林静,田莉,王虎. 针对低空微小型无人机的轻量型YOLOv5检测算法[J]. 光电子．激光, 2024, 35(6): 641-649 |
| |
| 作者姓名： | 魏峰 周建平 谭翔 林静 田莉 王虎 |
| |
| 作者单位： | 新疆大学 智能制造现代产业学院,新疆维吾尔自治区 乌鲁木齐 830000,新疆大学 智能制造现代产业学院,新疆维吾尔自治区 乌鲁木齐 830000,新疆大学 智能制造现代产业学院,新疆维吾尔自治区 乌鲁木齐 830000 ;中国科学院地理科学与资源研究所,北京 100101 ;中国科学院无人机应用与管控研究中心,北京 100101,中国科学院地理科学与资源研究所,北京 100101,中国科学院地理科学与资源研究所,北京 100101,新疆大学 智能制造现代产业学院,新疆维吾尔自治区 乌鲁木齐 830000 |
| |
| 基金项目： | 黑土地保护与利用科技创新工程专项资助(XDA28060400) 和中科吉安生态环境研究院院长基金(ZJIEES-2020-026)资助项目 |
| |
| 摘 要： | 针对低空微小型无人机对公共安全造成威胁的问题，本文基于YOLOv5(you only look once v5)网络提出了一种适用于移动端的轻量型目标检测模型YOLOv5＿SS。该模型以轻量型网络ShuffleNetv2替换YOLOv5原有的主干网络，引入SENet (squeeze-and-excitation networks)注意力机制，并采用Soft-NMS(soft non-maximum suppression)算法提升对密集重叠目标的检测效果。实验结果表明，该模型在数据集上对低空微小无人机进行检测的平均精确率均值(mean average precision@0.5,mAP50)为92.75%,精度为90.49%,参数量为0.237 4 M,浮点运算数为0.9千兆浮点运算(giga floating-point operations, GFLOPS)。具有检测精度高、内存占用率低的特点，有利于在移动终端上部署且在复杂背景及密集目标的场景下均有较好的检测效果。
|
| 关 键 词： | 无人机检测 深度学习 轻量型网络 注意力机制 非极大值抑制(NMS) |
| 收稿时间： | 2022-10-28 |
| 修稿时间： | 2023-02-09 |
Lightweight YOLOv5 detection algorithm for low-altitude micro UAV |
| |
| WEI Feng,ZHOU Jianping,TAN Xiang,LIN Jing,TIAN Li and WANG Hu. Lightweight YOLOv5 detection algorithm for low-altitude micro UAV[J]. Journal of Optoelectronics·laser, 2024, 35(6): 641-649 |
| |
| Authors: | WEI Feng ZHOU Jianping TAN Xiang LIN Jing TIAN Li WANG Hu |
| |
| Affiliation: | School of Intelligent Manufacturing Modern Industry,Xinjiang University,Urumchi, Xinjiang Uygur Autonomous Region 830000,China,School of Intelligent Manufacturing Modern Industry,Xinjiang University,Urumchi, Xinjiang Uygur Autonomous Region 830000,China,School of Intelligent Manufacturing Modern Industry,Xinjiang University,Urumchi, Xinjiang Uygur Autonomous Region 830000,China;Institute of Geographic Sciences and Natural Resources Research,Chinese Academy of Sciences,Beijing 100101,China;The Research Center for UAV Application and Regulation,Chinese Academy of Sciences,Beijing 100101,China,Institute of Geographic Sciences and Natural Resources Research,Chinese Academy of Sciences,Beijing 100101,China,Institute of Geographic Sciences and Natural Resources Research,Chinese Academy of Sciences,Beijing 100101,China and School of Intelligent Manufacturing Modern Industry,Xinjiang University,Urumchi, Xinjiang Uygur Autonomous Region 830000,China |
| |
| Abstract: | Aiming at the problem that low-altitude micro-UAVs pose a threat to public safety,this paper proposes a lightweight target detection model YOLOv5_SS suitable for mobile terminals based on the you only look once v5 (YOLOv5) network.In this model,the lightweight network ShuffleNetv2 replaces the original backbone network of YOLOv5,introduces squeeze-and-excitation networks (SENet) attention mechanism,and uses soft non-maximum suppression (Soft-NMS) algorithm to improve the detection effect of dense overlapping targets.The experimental results show that the mean average precision@0.5 (mAP50) of the model for the detection of low-altitude micro-UAV on the dataset is 92.75%,the accuracy is 90.49%,and the number of parameters is 0.237 4 M.The number of floating-point operations is 0.9GFLOPS (giga floating-point operations). |
| |
| Keywords: | UAV detection deep learning lightweight network attention mechanism non-maximum suppression (NMS) |
|
| 点击此处可从《光电子．激光》浏览原始摘要信息 |
|
点击此处可从《光电子．激光》下载全文 |
|
