基于改进YOLOv5s的轻量化船舶检测算法
Lightweight Ship Detection Algorithm Based on Improved YOLOv5s
张跃博 1郑元洲 2李果 2刘业余 2方正 2张远锋2
作者信息
- 1. 湖北省交通规划设计院股份有限公司 武汉 430063
- 2. 武汉理工大学航运学院 武汉 430063;内河航运技术湖北省重点实验室 武汉 430063
- 折叠
摘要
针对当前船舶检测模型复杂度高、对设备资源要求较高等问题,提出一种基于YOLOv5s的船舶目标检测算法SRE-YOLOv5s.该算法使用ShuffleNetV2轻量级网络替换YOLOv5s原始特征提取主干网络降低模型复杂度,并使用感受野模块(Receptive Field Block.RFB)增强算法特征处理能力,考虑到Complete IoU(CIoU)损失函数会影响模型收敛速度,采用Efficient IoU(EIoU)损失函数进行优化.通过对SRE-YOLOv5s算法进行训练和验证,结果表明:SRE-YOLOv5s在轻量化的同时保证了检测精度,与其他主流轻量化检测模型相比,SRE-YOLOv5s具有更好的检测性能.此外,根据船舶可视化检测结果可知,SRE-YOLOv5s模型在实际场景中具有较大的应用潜力.
Abstract
Aiming at the problems of high complexity of the current ship detection model and high re-quirements for equipment resources,a ship target detection algorithm SRE-YOLOv5s based on YOLOV5s was proposed.In this algorithm,ShuffleNetV2 lightweight network was used to replace YOLOv5s original feature extraction backbone network to reduce the model complexity,and RFB was used to enhance the feature processing ability of the algorithm.Considering that the loss function of Complete IoU(CIoU)will affect the convergence speed of the model,the loss function of Efficient IoU(EIoU)was adopted for optimization.By training and verifying the SRE-YOLOv5s algorithm,the results show that SRE-YOLOv5s is lightweight and ensures the detection accuracy.Compared with other mainstream lightweight detection models,SRE-YOLOv5s has better detection performance.In addition,according to the visual inspection results of the ship,the SRE-YOLOv5s model has great ap-plication potential in the actual scene.
关键词
船舶检测/YOLOv5s/轻量级网络/感受野Key words
ship detection/YOLOv5s/lightweight network/receptive field引用本文复制引用
基金项目
国家自然科学基金(51979215)
国家自然科学基金(52171350)
出版年
2024
NETL
NSTL
万方数据