A Lightweight Model for Real-Time Monitoring of Ships

被引:22
作者
Xing, Bowen [1 ]
Wang, Wei [1 ]
Qian, Jingyi [2 ,3 ]
Pan, Chengwu [4 ]
Le, Qibo [4 ]
机构
[1] Shanghai Ocean Univ, Coll Engn Sci & Technol, Shanghai 201306, Peoples R China
[2] Tongji Univ, Coll Elect & Informat Engn, Shanghai 201804, Peoples R China
[3] Shanghai Aerosp Elect Co Ltd, Shanghai 201800, Peoples R China
[4] Ningbo Commun Ctr, Ningbo 315800, Peoples R China
关键词
ship monitoring; deep learning; lightweight model; real-time tracking; YOLOv8; NETWORK;
D O I
10.3390/electronics12183804
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Real-time monitoring of ships is crucial for inland navigation management. Under complex conditions, it is difficult to balance accuracy, real-time performance, and practicality in ship detection and tracking. We propose a lightweight model, YOLOv8-FAS, to address this issue for real-time ship detection and tracking. First, FasterNet and the attention mechanism are integrated and introduced to achieve feature extraction simply and efficiently. Second, the lightweight GSConv convolution method and a one-shot aggregation module are introduced to construct an efficient network neck to enhance feature extraction and fusion. Furthermore, the loss function is improved based on ship characteristics to make the model more suitable for ship datasets. Finally, the advanced Bytetrack tracke is added to achieve the real-time detection and tracking of ship targets. Compared to the YOLOv8 model, YOLOv8-FAS reduces computational complexity by 0.8x109 terms of FLOPs and reduces model parameters by 20%, resulting in only 2.4x106 parameters. The mAP-0.5 is improved by 0.9%, reaching 98.50%, and the real-time object tracking precision of the model surpasses 88%. The YOLOv8-FAS model combines light weight with high precision, and can accurately perform ship detection and tracking tasks in real time. Moreover, it is suitable for deployment on hardware resource-limited devices such as unmanned surface ships.
引用
收藏
页数:17
相关论文
共 39 条
[1]  
Bauwens Jef, 2023, Proceedings of PIANC Smart Rivers 2022: Green Waterways and Sustainable Navigations. Lecture Notes in Civil Engineering (264), P1353, DOI 10.1007/978-981-19-6138-0_119
[2]  
Bochkovskiy A, 2020, Arxiv, DOI arXiv:2004.10934
[3]   Run, Don't Walk: Chasing Higher FLOPS for Faster Neural Networks [J].
Chen, Jierun ;
Kao, Shiu-Hong ;
He, Hao ;
Zhuo, Weipeng ;
Wen, Song ;
Lee, Chul-Ho ;
Chan, S. -H. Gary .
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, :12021-12031
[4]   Convolutional Social Pooling for Vehicle Trajectory Prediction [J].
Deo, Nachiket ;
Trivedi, Mohan M. .
PROCEEDINGS 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2018, :1549-1557
[5]   Ship detection with deep learning: a survey [J].
Er, Meng Joo ;
Zhang, Yani ;
Chen, Jie ;
Gao, Wenxiao .
ARTIFICIAL INTELLIGENCE REVIEW, 2023, 56 (10) :11825-11865
[6]  
Howard AG, 2017, Arxiv, DOI arXiv:1704.04861
[7]  
Ge Z, 2021, Arxiv, DOI [arXiv:2107.08430, 10.48550/arXiv.2107.08430, DOI 10.48550/ARXIV.2107.08430]
[8]   GhostNet: More Features from Cheap Operations [J].
Han, Kai ;
Wang, Yunhe ;
Tian, Qi ;
Guo, Jianyuan ;
Xu, Chunjing ;
Xu, Chang .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1577-1586
[9]  
Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]
[10]   A Survey of Deep Learning-Based Object Detection [J].
Jiao, Licheng ;
Zhang, Fan ;
Liu, Fang ;
Yang, Shuyuan ;
Li, Lingling ;
Feng, Zhixi ;
Qu, Rong .
IEEE ACCESS, 2019, 7 :128837-128868