基于轻量化YOLOv8n的船舶号灯识别

展开
  •  (大连海事大学 航海学院,辽宁 大连 116026) 
乔亚静(2001-),女,硕士生,研究方向:船舶号灯识别。E-mail:2276925216@qq.com; 赵月林*(1966-),男,博士,教授,研究方向:船舶操纵及避碰。E-mail:13304289000@163.com

网络出版日期: 2025-01-07

基金资助

辽宁省“兴辽英才计划”项目(XLYC1902071);智能船舶安全航行岸基监测预警关键技术研究

Ship's lights Identification Based on Lightweight YOLOv8n

Expand
  • (Navigation College, Dalian Maritime University, Dalian 116026, China)

Online published: 2025-01-07

摘要

针对船舶号灯目标检测中参数计算量大、背景光线复杂及号灯种类繁多等问题,对YOLOv8n做出了改进,以满足实时准确识别船舶号灯的要求。首先使用VanillaNet网络作为主干特征提取网络,以降低模型的计算成本,满足实时检测要求;其次引入颜色注意力模块以确定和增强号灯的颜色特征,以提高在复杂光线背景下号灯颜色的识别能力;再次,针对号灯的大小、频率、各视图空间排列等特征,构建MoE-layer模块代替C2f模块,以提高船舶号灯识别的准确性;最后通过Focal Loss调节对难易分类样本的关注来解决类别不平衡问题,提高模型对号灯目标检测能力。实验结果表明,相较于原基线YOLOv8n模型,改进后模型在参数量和计算量分别降低37.7%和52.8%的情况下,精度和mAP@0.5分别提升3.3%和2.2%,达到98.3%和98.7%,可以满足实时识别船舶号灯的要求。

本文引用格式

乔亚静, 高祥雨, 赵月林 . 基于轻量化YOLOv8n的船舶号灯识别[J]. 大连海事大学学报, 2025 , 51(1) : 82 -91 . DOI: 10.16411/j.cnki.issn1006-7736.2025.01.009

Abstract

This study addresses the challenges associated with high computational parameter volume, intricate complex background lighting, and the diversity of the ship's lights in the context of target detection. improvements have been made to YOLOv8n to meet the requirements for real-time and accurate identification of ship’s lights. Initially, we employ the VanillaNet as the core feature extraction network, which significantly diminishes the model's computational overhead and ensures compliance with real-time detection demands. Subsequently, we integrate a color attention module designed to discern and amplify the distinctive color attributes of the navigation lights, thereby bolstering their recognition even amidst complex lighting conditions. Furthermore, to accommodate the unique characteristics of the ship's lights, including their size, frequency, and spatial distribution across different perspectives, introduce a Mixture of Experts (MoE)-layer module as a substitute for the conventional C2f module, further refining the identification accuracy. Finally, the Focal Loss function is adopted to adjust the focus on easy-to-hard classified samples, addressing the class imbalance issue and improving the model's ability to detect ship’s lights. Experimental results demonstrate that, compared to the original baseline model YOLOv8n, the improved model reduces parameters and computation by 37.7% and 52.8%, respectively, while increasing accuracy and mAP@0.5 by 3.3% and 2.2% to 98.3% and 98.7%, respectively, and therefore the improved YOLOv8n meets the requirements for real-time identification of ship’s lights.

参考文献

[1] 吴兆麟, 赵月林. 船舶避碰与值班(第5版) [M]. 大连: 大连海事大学出版社, 2021.
WU, Z, L, ZHAO, Y, L. Collision avoidance and watchkeeping(5thEdition)[M]. Dalian: Dalian Maritime University Press, 2021. (in Chinese)
[2] 胡鑫. 基于视频分析的海上航行船舶号灯识别方法 [D]. 大连: 大连海事大学, 2021. 
HU X. Recognition method of marine navigation ship lights based on video analysis[D]. Dalian: Dalian Maritime University, 2021. (in Chinese)
[3] 熊蕾, 金欢. 利用BP神经网络进行船舶号灯识别 [J].舰船科学技术, 2017, 39(4): 175-177.
XIONG L, JIN H. Application of BP neural network on identification of shipment navigation lights[J]. Ship science and technology, 2017, 39(4): 175-177. (in Chinese)
[4] 朱金善, 孙立成, 胡江强, 等. 基于克隆优化的船舶号灯神经网络识别模型 [J].大连海事大学学报, 2015, 41(2): 41-45.
ZHU J S, SUN L C, HU J Q, et al. Neural network recognition model of ship lights based on clonal optimization[J]. Journal of Dalian Maritime University, 2015, 41(2): 41-45. (in Chinese)
[5] GAO Y, WU Z, REN M, et al. Improved YOLOv4 Based on Attention Mechanism for Ship Detection in SAR Images[J]. IEEE Access, 2022,1023785-23797. 
[6] 张晓鹏, 许志远, 曲胜, 等. 基于改进YOLOv5深度学习的海上船舶识别算法 [J]. 大连海洋大学学报, 2022, 37(5): 866-872.
ZHNAG X P, XU Z Y, QU S, et al. Marine ship recognition algorithm based on improved YOLOv5 deep learning[J]. Journal of Dalian Ocean University, 2022, 37(5): 866-872. (in Chinese)
[7] GIRSHICK R. Fast r-cnn[C]//Proceedings of the IEEE international conference on computer vision.2015: 1440-1448.
[8] XU C A, SU H, GAO L, et al. Feature aligned ship detection based on improved RPDet in SAR images[J]. Displays: Technology and Applications,2022,74. 102191.
[9] 王文亮, 李延祥, 张一帆, 等. MPANet-YOLOv5: 多路径聚合网络复杂海域目标检测 [J]. 湖南大学学报(自然科学版), 2022, 49(10): 69-76.
WANG W L, LI Y X, ZHANG Y F, et al. MPANet-YOLOv5: multipath aggregation network for complex sea area target detection[J]. Journal of Hunan University(Natural Science Edition), 2022 49(10): 69-76. (in Chinese) 
[10]丘锐聪, 周海峰, 陈颖, 等. 基于轻量化YOLOv7-tiny的船舶目标检测算法 [J]. 大连海事大学学报, 2024, 50(2): 31-40.
QIU R C, ZHOU H F, CHEN Y, et al. Ship target detection algorithm based on lightweight YOLOv7-tiny [J]. Journal of Dalian Maritime University, 2024, 50(2): 31-40. (in Chinese)
[11]T. NISHINA, E. SHIMIZU. A Preliminary Study on Obstacle Detection System for Night Navigation[C]. //2020 IEEE/SICE International Symposium on System Integration: SII 2020, Honolulu, Hawaii, USA, 12-15 January 2020, [v.2].: Institute of Electrical and Electronics Engineers, 2020: 1107-1112.
[12]MA N, SU Y, YANG L, et al. Wheat Seed Detection and Counting Method Based on Improved YOLOv8 Model[J].Sensors, 2024, 24(5):1654.
[13]CHEN H, WANG Y, GUO J, et al. VanillaNet: the Power of minimalism in deep learning[J]. arXiv, 2023. arXiv:2305.12972v2. 
[14] 刘珂琪, 董绵绵, 王鹏, 等. 一种基于图像增强的交通信号灯识别方法 [J]. 电子测量技术, 2022, 45(07): 137-145. 
LIU K Q, DONG M M, WANG P, et al. A traffic light recognition method based on image enhancement[J]. electronic measurement technology, 2022, 45(07): 137-145. (in Chinese)
[15] YUN S, JEONG; M, KIM R, et al. Graph transformer networks[C]//Advances in Neural Information Processing Systems. Vancouver: NIPS, 2019: 11983-11993.
[16] 徐胜军, 欧阳朴衍, 郭学源, 等. 多尺度特征融合空洞卷积ResNet遥感图像建筑物分割 [J]. 光学精密工程, 2020, 28(7):1588-1599.
XU S J, OUYANG P Y, GUO X Y, et al. Building segmentation in remote sensing image based on multiscale-feature fusion dilated convolution resnet[J]. Optical Precision Engineering, 2020, 28(7): 1588-1599. (in Chinese)
[17] LEE-J, AINSLIE J, ECKSTEIN I, et al. FNet: Mixing Tokens with Fourier Transforms[J]. arXiv, 2021, arXiv:2105.03824.
[18] LIN T Y, GOYAL P, GIRSHICK R, et al. Focal Loss for Dense Object Detection[J]. arXiv, 2018, arXiv:1708.0202.
[19] REDMON J,FARHADI A. Yolov3: an incremental improvement[C]//EEE Conference on Computer Vision and Pattern Recognition.[S.l.:s.n.], 2018: 1804.02767.
[20] CARION N., MASSA F, SYNNAEVE G, et al. End-to-end object detection with transformers[J]. arXiv, 2020, arXiv:2005.12872. 
[21] 马鸽, 李洪伟, 严梓维, 等. 基于多注意力的改进YOLOv5s小目标检测算法 [J]. 工程科学学报, 2024, 46(9): 1647-1658.
MA G, LI H W, YAN Z W, et al. Improved small target detection algorithm based on multiattention and YOLOv5s for traffic sign recognition[J]. The Journal of Engineering Science, 2024, 46(9): 1647-1658. (in Chinese)
[22] LI C Y, LI L, JIANG H L, et al. YOLOv6: A Single-Stage Object Detection Framework for Industrial Applications [J]. arXiv: 2209.02976, 2022.
[23] 张泽, 张建权, 周国鹏. 基于改进YOLOv8s的摄像头模组缺陷检测 [J]. 计算机与现代化, 2024(9): 107-113. 
ZHANG Z, ZHANG J Q, ZHOU G P. Camera module defect based on improved YOLOv8s[J]. Computer and Modernization, 2024(9): 107-113. (in Chinese)

文章导航

/