Ship's lights Identification Based on Lightweight YOLOv8n

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

Online published: 2025-01-07

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.

Cite this article

QIAO Yajing, GAO Xiangyu , ZHAO Yuelin . Ship's lights Identification Based on Lightweight YOLOv8n[J]. Journal of Dalian Maritime University, 2025 , 51(1) : 82 -91 . DOI: 10.16411/j.cnki.issn1006-7736.2025.01.009

References

[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)

Outlines

/