Skip to main navigation Skip to search Skip to main content

A ship detection method based on improved YOLOv8 models and ensemble learning

  • Beihang University

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

To solve the challenges of reduced accuracy and robustness in maritime ship detection using a single detector, particularly in complex environments with fog interference, an ensemble learning method based on two improved YOLOv8 models is proposed. The YOLOv8 architecture is enhanced by incorporating the Coordinate Attention mechanism to improve multi-scale detection performance, and replacing the CIOU loss function with WIoU to enhance accuracy. Two independently trained YOLOv8 models are employed as base detectors, each optimized for ship detection in either foggy or clear conditions. A coordinate-weighted algorithm merges outputs from the two detectors, using ensemble learning to enhance robustness in foggy and clear conditions. A dataset of 5, 546 images, divided into foggy and clear subsets, was created and expanded through data augmentation. Experimental results demonstrate detection accuracies of 85.9% and 92.1% with recall rates of 93.0% and 95.1% for the proposed method.

Original languageEnglish
Title of host publicationTenth Symposium on Novel Optoelectronic Detection Technology and Applications
EditorsChen Ping
PublisherSPIE
ISBN (Electronic)9781510688148
DOIs
StatePublished - 2025
Event10th Symposium on Novel Optoelectronic Detection Technology and Applications - Taiyuan, China
Duration: 1 Nov 20243 Nov 2024

Publication series

NameProceedings of SPIE - The International Society for Optical Engineering
Volume13511
ISSN (Print)0277-786X
ISSN (Electronic)1996-756X

Conference

Conference10th Symposium on Novel Optoelectronic Detection Technology and Applications
Country/TerritoryChina
CityTaiyuan
Period1/11/243/11/24

Keywords

  • computer vision
  • convolutional neural networks
  • deep learning
  • ensemble learning
  • remote sensing

Fingerprint

Dive into the research topics of 'A ship detection method based on improved YOLOv8 models and ensemble learning'. Together they form a unique fingerprint.

Cite this