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

An Improved YOLOv5 Approach for Multi-object Detection of Unmanned Surface Vehicles

  • Ying Yang,
  • Changyun Wei,
  • Qi Liu,
  • Shouqian Li

摘要

This paper aims to address the challenge of multi-object detection for unmanned surface vehicles (USVs) due to the partially occluded and blurred nature of water surface targets in captured images. To this end, we propose an enhanced version of the YOLOv5 network to improve detection accuracy and efficiency. Firstly, the Ghost module is introduced to reduce the computational complexity when obtaining feature maps, then the SE module is introduced to enhance the extraction of feature informations, and finally the loss function of YOLOv5 is optimized by CIOU to improve the training effect of the model. Results of our experiments indicate that the proposed YOLOv5s-GS model is lightweight and has significantly improved detection accuracy and efficiency over the original YOLOv5s for multi-object detection tasks.