Open Access

ARTICLE

Underwater Objects Detection Based on a Multi-Stage Deep Learning Framework

Rana Lateef¹, Asmaa Abdul Jabbar^2,*

1 Department of Cybersecurity Science, College of Science, Al-Iraqia Science University, Baghdad, Iraq
2 Department of Computer Science, College of Science, Mustansiriyah University, Baghdad, Iraq

* Corresponding Author: Asmaa Abdul Jabbar. Email:

(This article belongs to the Special Issue: Advances in Object Detection and Recognition)

Computers, Materials & Continua 2026, 88(2), 86 https://doi.org/10.32604/cmc.2026.080975

Received 20 February 2026; Accepted 29 April 2026; Issue published 15 June 2026

Abstract

The challenges of underwater object detection are derived from complex environmental conditions, including light scattering, absorption, and turbidity. The deep learning approaches have enhanced the detection of objects in these low-visual conditions. This work presents a multi-stage object-detection framework for the underwater environment that performs well on the Semantic Segmentation of Underwater Imagery (SUIM) benchmark. To begin with, there is the adaptive Multi-Scale Retinex with Color Restoration (MSRCR) algorithm, which improves image quality by correcting color distortions and increasing contrast. Second, an augmented YOLOv8 model (with a ResNet-50 backbone and the Convolutional Block Attention Module (CBAM)) is used to extract powerful features for object detection in low-light conditions. Lastly, a LightGBM classifier selects initial detections using contextual information to reduce false positives. The proposed model is evaluated on the SUIM dataset, with ground-truth segmentation masks converted to bounding boxes according to standard COCO protocols for detection-based training and evaluation. Comparative experiments against reimplemented YOLO-based underwater detectors demonstrate that the proposed model achieves a macro-average mAP@0.5 of 86.33%, outperforming YOLOv8-nano (80.13%), YOLOv7+enhancement (84.23%), and AIT-YOLOv7 (83.4%) on the SUIM benchmark under similar conditions.

---

Keywords

---