<p>Images of military targets typically exhibit characteristics such as camouflage, varying scales, and uneven distribution, making target detection complex and challenging. Furthermore, the limited computing resources of unmanned platforms, such as drones and ground unmanned vehicles, make it difficult to deploy detectors with large parameter counts on these systems. In this study, we enhanced the components of YOLOv8n and proposed a new lightweight military target detection algorithm named YOLO-E. We constructed a military target dataset composed of armed personnel holding various weapons to facilitate the verification of the different algorithms. In our proposed algorithm, we applied an Efficient Multi-Scale Convolution module in the feature extraction network to improve the detection speed of military targets. Additionally, we designed a head network based on weight sharing, which significantly reduced the model parameters. We also propose a novel bounding box loss function, the Normalized Corner Distance IoU, to further enhance the detection accuracy of military targets. We tested YOLO-E on a self-developed military target dataset. Experimental results showed that compared to the original YOLOv8n algorithm, YOLO-E improved the detection accuracy by approximately 2.33%, increased detection speed by 1.68%, reduced parameters by 30.87%, and decreased computational complexity by 37.33%. Furthermore, we compared our method with several advanced object detection algorithms. The results demonstrated that YOLO-E outperformed the others in terms of comprehensive parameters, real-time performance, and accuracy. The proposed network model provides an effective auxiliary support for analyzing battlefield situations.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

YOLO-E: a lightweight object detection algorithm for military targets

  • Yong Sun,
  • Jianzhong Wang,
  • Yu You,
  • Zibo Yu,
  • Shaobo Bian,
  • Endi Wang,
  • Weichao Wu

摘要

Images of military targets typically exhibit characteristics such as camouflage, varying scales, and uneven distribution, making target detection complex and challenging. Furthermore, the limited computing resources of unmanned platforms, such as drones and ground unmanned vehicles, make it difficult to deploy detectors with large parameter counts on these systems. In this study, we enhanced the components of YOLOv8n and proposed a new lightweight military target detection algorithm named YOLO-E. We constructed a military target dataset composed of armed personnel holding various weapons to facilitate the verification of the different algorithms. In our proposed algorithm, we applied an Efficient Multi-Scale Convolution module in the feature extraction network to improve the detection speed of military targets. Additionally, we designed a head network based on weight sharing, which significantly reduced the model parameters. We also propose a novel bounding box loss function, the Normalized Corner Distance IoU, to further enhance the detection accuracy of military targets. We tested YOLO-E on a self-developed military target dataset. Experimental results showed that compared to the original YOLOv8n algorithm, YOLO-E improved the detection accuracy by approximately 2.33%, increased detection speed by 1.68%, reduced parameters by 30.87%, and decreased computational complexity by 37.33%. Furthermore, we compared our method with several advanced object detection algorithms. The results demonstrated that YOLO-E outperformed the others in terms of comprehensive parameters, real-time performance, and accuracy. The proposed network model provides an effective auxiliary support for analyzing battlefield situations.