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 条
[11]   Ship Detection and Tracking in Inland Waterways Using Improved YOLOv3 and Deep SORT [J].
Jie, Yang ;
Leonidas, LilianAsimwe ;
Mumtaz, Farhan ;
Ali, Munsif .
SYMMETRY-BASEL, 2021, 13 (02) :1-19
[12]  
Jocher G., 2022, Zenodo
[13]  
Li CY, 2022, Arxiv, DOI [arXiv:2209.02976, 10.48550/arXiv.2209.02976]
[14]  
Li HL, 2024, Arxiv, DOI arXiv:2206.02424
[15]   Deep Learning for SAR Ship Detection: Past, Present and Future [J].
Li, Jianwei ;
Xu, Congan ;
Su, Hang ;
Gao, Long ;
Wang, Taoyang .
REMOTE SENSING, 2022, 14 (11)
[16]  
Li X., 2020, Adv Neural Inf Process Syst, V33, P21002, DOI 10.5555/3495724.3497487
[17]   Feature Pyramid Networks for Object Detection [J].
Lin, Tsung-Yi ;
Dollar, Piotr ;
Girshick, Ross ;
He, Kaiming ;
Hariharan, Bharath ;
Belongie, Serge .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :936-944
[18]   Squeeze and Excitation Rank Faster R-CNN for Ship Detection in SAR Images [J].
Lin, Zhao ;
Ji, Kefeng ;
Leng, Xiangguang ;
Kuang, Gangyao .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2019, 16 (05) :751-755
[19]   Path Aggregation Network for Instance Segmentation [J].
Liu, Shu ;
Qi, Lu ;
Qin, Haifang ;
Shi, Jianping ;
Jia, Jiaya .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :8759-8768
[20]   SSD: Single Shot MultiBox Detector [J].
Liu, Wei ;
Anguelov, Dragomir ;
Erhan, Dumitru ;
Szegedy, Christian ;
Reed, Scott ;
Fu, Cheng-Yang ;
Berg, Alexander C. .
COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 :21-37