Abstract:Underwater object detection and classification pose significant challenges due to environmental factors such as water turbidity and variable lighting conditions. This research proposes a novel approach that integrates advanced imaging techniques with diffusion models to address these challenges effectively, aligning with Sustainable Development Goal (SDG) 14: Life Below Water. The methodology leverages the Convolutional Block Attention Module (CBAM), Modified Swin Transformer Block (MSTB), and Diffusion model to enhance the quality of underwater images, thereby improving the accuracy of object detection and classification tasks. This study utilizes the TrashCan dataset, comprising diverse underwater scenes and objects, to validate the proposed method's efficacy. This study proposes an advanced imaging technique YOLO (you only look once) network (AIT-YOLOv7) for detecting objects in underwater images. This network uses a modified U-Net, which focuses on informative features using a convolutional block channel and spatial attentions for color correction and a modified swin transformer block for resolution enhancement. A novel diffusion model proposed using modified U-Net with ResNet understands the intricate structures in images with underwater objects, which enhances detection capabilities under challenging visual conditions. Thus, AIT-YOLOv7 net precisely detects and classifies different classes of objects present in this dataset. These improvements are crucial for applications in marine ecology research, underwater archeology, and environmental monitoring, where precise identification of marine debris, biological organisms, and submerged artifacts is essential. The proposed framework advances underwater imaging technology and supports the sustainable management of marine resources and conservation efforts. The experimental results demonstrate that state-of-the-art object detection methods, namely SSD, YOLOv3, YOLOv4, and YOLOTrashCan, achieve mean accuracies (mAP@0.5) of 57.19%, 58.12%, 59.78%, and 65.01%, respectively, whereas the proposed AIT-YOLOv7 net reaches a mean accuracy (mAP@0.5) of 81.4% on the TrashCan dataset, showing a 16.39% improvement. Due to this improvement in the accuracy and efficiency of underwater object detection, this research contributes to broader marine science and technology efforts, promoting the better understanding and management of aquatic ecosystems and helping to prevent and reduce the marine pollution, as emphasized in SDG 14.

A novel finetuned YOLOv8 model for real-time underwater trash detection

Research on Underwater Small Target Detection Algorithm Based on Improved YOLOv3

YOLOv8-C2f-Faster-EMA: An Improved Underwater Trash Detection Model Based on YOLOv8

Underwater Object Detection Based on Enhanced YOLO

Enhancing Underwater Object Detection and Classification Using Advanced Imaging Techniques: A Novel Approach with Diffusion Models

A real-time object detection method for underwater complex environments based on FasterNet-YOLOv7

RTL-YOLOv8n: A Lightweight Model for Efficient and Accurate Underwater Target Detection

Improved YOLOv7 model for underwater sonar image object detection

A lightweight YOLOv8 integrating FasterNet for real-time underwater object detection

Underwater Target Detection by Side-Scan Sonar Based on Yolov7-Attention

YOLOv8-MU: An Improved YOLOv8 Underwater Detector Based on a Large Kernel Block and a Multi-Branch Reparameterization Module

Attention-Based Lightweight YOLOv8 Underwater Target Recognition Algorithm

A Lightweight Underwater Object Detection Algorithm with Adaptive Image Enhancement Based on YOLOv8

Underwater Object Detection Algorithm Based on an Improved YOLOv8

YOLO‐UOD: An underwater small object detector via improved efficient layer aggregation network

Image-Fused-Guided Underwater Object Detection Model Based on Improved YOLOv7

An Improved Yolo Algorithm for Fast and Accurate Underwater Object Detection

FEB-YOLOv8: A multi-scale lightweight detection model for underwater object detection

Underwater small target detection under YOLOv8-LA model

Underwater object detection algorithm based on attention mechanism and cross-stage partial fast spatial pyramidal pooling

YOLOv7-SN: Underwater Target Detection Algorithm Based on Improved YOLOv7