Improved Ship Detection with YOLOv8 Enhanced with MobileViT and GSConv
In tasks that require ship detection and recognition, the irregular shapes of ships and complex backgrounds pose significant challenges. This paper presents an advanced extension of the YOLOv8 model to address these challenges. A lightweight visual transformer, MobileViTSF, is proposed and combined...
Main Authors: | , |
---|---|
Format: | Article |
Language: | English |
Published: |
MDPI AG
2023-11-01
|
Series: | Electronics |
Subjects: | |
Online Access: | https://www.mdpi.com/2079-9292/12/22/4666 |
_version_ | 1797459497348759552 |
---|---|
author | Xuemeng Zhao Yinglei Song |
author_facet | Xuemeng Zhao Yinglei Song |
author_sort | Xuemeng Zhao |
collection | DOAJ |
description | In tasks that require ship detection and recognition, the irregular shapes of ships and complex backgrounds pose significant challenges. This paper presents an advanced extension of the YOLOv8 model to address these challenges. A lightweight visual transformer, MobileViTSF, is proposed and combined with the YOLOv8 model. To address the loss of semantic information that arises from inconsistent scales in the detection of small ships, a layer intended for the detection of small targets is introduced to lead to improved fusion of deep and shallow features. Furthermore, the traditional convolution (Conv) blocks are replaced with GSConv blocks, and a novel GSC2f block is designed for fewer model parameters and improved detection performance. Experiments on a benchmark dataset suggest that this new model can achieve significantly improved accuracy for ship detection with fewer model parameters and a reduced model size. A comparison with several other state-of-the-art methods shows that higher accuracy can be obtained for ship detection with this model. Moreover, this new model is suitable for edge computing devices, demonstrating practical application value. |
first_indexed | 2024-03-09T16:53:11Z |
format | Article |
id | doaj.art-d8f690dc3a9a4b859be3fd8b34fb38a4 |
institution | Directory Open Access Journal |
issn | 2079-9292 |
language | English |
last_indexed | 2024-03-09T16:53:11Z |
publishDate | 2023-11-01 |
publisher | MDPI AG |
record_format | Article |
series | Electronics |
spelling | doaj.art-d8f690dc3a9a4b859be3fd8b34fb38a42023-11-24T14:39:35ZengMDPI AGElectronics2079-92922023-11-011222466610.3390/electronics12224666Improved Ship Detection with YOLOv8 Enhanced with MobileViT and GSConvXuemeng Zhao0Yinglei Song1School of Automation, Jiangsu University of Science and Technology, Zhenjiang 212003, ChinaSchool of Automation, Jiangsu University of Science and Technology, Zhenjiang 212003, ChinaIn tasks that require ship detection and recognition, the irregular shapes of ships and complex backgrounds pose significant challenges. This paper presents an advanced extension of the YOLOv8 model to address these challenges. A lightweight visual transformer, MobileViTSF, is proposed and combined with the YOLOv8 model. To address the loss of semantic information that arises from inconsistent scales in the detection of small ships, a layer intended for the detection of small targets is introduced to lead to improved fusion of deep and shallow features. Furthermore, the traditional convolution (Conv) blocks are replaced with GSConv blocks, and a novel GSC2f block is designed for fewer model parameters and improved detection performance. Experiments on a benchmark dataset suggest that this new model can achieve significantly improved accuracy for ship detection with fewer model parameters and a reduced model size. A comparison with several other state-of-the-art methods shows that higher accuracy can be obtained for ship detection with this model. Moreover, this new model is suitable for edge computing devices, demonstrating practical application value.https://www.mdpi.com/2079-9292/12/22/4666ship detectionobject detectionYOLOv8MobileViTGSConv |
spellingShingle | Xuemeng Zhao Yinglei Song Improved Ship Detection with YOLOv8 Enhanced with MobileViT and GSConv Electronics ship detection object detection YOLOv8 MobileViT GSConv |
title | Improved Ship Detection with YOLOv8 Enhanced with MobileViT and GSConv |
title_full | Improved Ship Detection with YOLOv8 Enhanced with MobileViT and GSConv |
title_fullStr | Improved Ship Detection with YOLOv8 Enhanced with MobileViT and GSConv |
title_full_unstemmed | Improved Ship Detection with YOLOv8 Enhanced with MobileViT and GSConv |
title_short | Improved Ship Detection with YOLOv8 Enhanced with MobileViT and GSConv |
title_sort | improved ship detection with yolov8 enhanced with mobilevit and gsconv |
topic | ship detection object detection YOLOv8 MobileViT GSConv |
url | https://www.mdpi.com/2079-9292/12/22/4666 |
work_keys_str_mv | AT xuemengzhao improvedshipdetectionwithyolov8enhancedwithmobilevitandgsconv AT yingleisong improvedshipdetectionwithyolov8enhancedwithmobilevitandgsconv |