Research on marine flexible biological target detection based on improved YOLOv8 algorithm

• Published in: PeerJ. Computer science 2024-08, Vol.10, p.e2271, Article e2271
• Main Authors: Tian, Yu, Liu, Yanwen, Lin, Baohang, Li, Peng
• Format: Article
• Language: English
• Subjects: Algorithms; Algorithms and Analysis of Algorithms; Artificial Intelligence; CLAHE; Computer Vision; Data Mining and Machine Learning; Improved YOLOv8; Marine flexible biological targets; Neural Networks; Target detection
• ISSN: 2376-5992; 2376-5992
• DOI: 10.7717/peerj-cs.2271

To address the challenge of suboptimal object detection outcomes stemming from the deformability of marine flexible biological entities, this study introduces an algorithm tailored for detecting marine flexible biological targets. Initially, we compiled a dataset comprising marine flexible biological subjects and developed a Contrast Limited Adaptive Histogram Equalization (CLAHE) algorithm, supplemented with a boundary detection enhancement module, to refine underwater image quality and accentuate the distinction between the images' foregrounds and backgrounds. This enhancement mitigates the issue of foreground-background similarity encountered in detecting marine flexible biological entities. Moreover, the proposed adaptation incorporates a Deformable Convolutional Network (DCN) network module in lieu of the C2f module within the YOLOv8n algorithm framework, thereby augmenting the model's proficiency in capturing geometric transformations and concentrating on pivotal areas. The Neck network module is enhanced with the RepBi-PAN architecture, bolstering its capability to amalgamate and emphasize essential characteristics of flexible biological targets. To advance the model's feature information processing efficiency, we integrated the SimAM attention mechanism. Finally, to diminish the adverse effects of inferior-quality labels within the dataset, we advocate the use of WIoU (Wise-IoU) as a bounding box loss function, which serves to refine the anchor boxes' quality assessment. Simulation experiments show that, in comparison to the conventional YOLOv8n algorithm, our method markedly elevates the precision of marine flexible biological target detection.