基于轻量化YOLOv7-tiny的船舶目标检测算法

丘锐聪, 周海峰, 陈颖, 张兴杰, 黄金满, 翁卫征

PDF(16642 KB)
PDF(16642 KB)
大连海事大学学报 ›› 2024, Vol. 50 ›› Issue (02) : 31-40. DOI: 10.16411/j.cnki.issn1006-7736.2024.02.004

基于轻量化YOLOv7-tiny的船舶目标检测算法

  • 丘锐聪, 周海峰, 陈颖, 张兴杰, 黄金满, 翁卫征
作者信息 +
History +

摘要

为解决船舶目标检测算法参数量与计算量大,以及受内河环境下近岸复杂背景影响和船舶相互遮挡导致船舶检测困难的问题,基于YOLOv7-tiny做出改进,提出MED-YOLO船舶目标检测轻量化算法。首先,使用MobileNetV3网络作为主干特征提取网络,极大地降低了模型计算成本;其次,将EMA注意力模块引入颈部网络,构建EMA-ELAN模块,增强网络多维度感知和多尺度特征提取能力;然后,选用将尺度感知、空间感知和任务感知三合一的Dyhead作为改进模型的检测头部,以获得更强的特征表达能力;最后,使用具有动态非单调聚焦机制的WIoU作为模型边界框损失函数,提高模型应对船舶遮挡的能力,提升检测性能。实验结果表明,MED-YOLO相较YOLOv7-tiny在参数量与计算量方面分别减少39.8%和55.0%,精度与mAP@0.5分别提高了1.4%和1.0%,达到98.3%和98.9%,在实现轻量化的同时具有更好的检测性能,满足了计算资源受限环境下的部署需求,具有一定的工程实际意义。

关键词

YOLOv7-tiny / 目标检测 / 轻量化 / 注意力机制 / 损失函数

中图分类号

U675.79 / TP183 / TP391.41

引用本文

导出引用
丘锐聪, 周海峰, 陈颖, 张兴杰, 黄金满, 翁卫征. 基于轻量化YOLOv7-tiny的船舶目标检测算法. 大连海事大学学报. 2024, 50(02): 31-40 https://doi.org/10.16411/j.cnki.issn1006-7736.2024.02.004

基金

国家自然科学基金资助项目(51179074); 福建省自然科学基金资助项目(2021J01839); 集美大学安麦信产学研项目(S20127)

评论

PDF(16642 KB)

Accesses

Citation

Detail

段落导航
相关文章

/