Underwater image enhancement and detection based on convolutional DCP and YOLOv5

Guodong Liu, Lihui Feng*, Jihua Lu, Lei Yan

*Corresponding author for this work

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

1 Citation (Scopus)

Abstract

Underwater image restoration is conducive to better underwater resource detection and information effective trans-mission. However, the light in the complex water body diffusely reflective and the selection absorption of different band light results in blurring and color distortion of underwater image. Therefore, we propose a convolutional Dark Channel Prior (DCP) underwater image recovery algorithm to enhance underwater images. It can do the pre-processing work for the subsequent YOLOv5 object recognition. The enhancement algorithm first performs Commission International Eclairage Lab (CIELAB) equalization of underwater images for color distortion correction. Meanwhile, the underwater image formation parameters are estimated by the minimum convolution region DCP. Then, Contrast Limited Adaptive Histogram Equalization (CLAHE) is per-formed to obtain an enhanced underwater image. Finally, the enhanced underwater image is input to the YOLOv5 model for detection. Experimental results show that the proposed method outperforms state-of-art algorithms in terms of image recovery effect, evaluation quality and detection accuracy.

Original languageEnglish
Title of host publicationProceedings of the 41st Chinese Control Conference, CCC 2022
EditorsZhijun Li, Jian Sun
PublisherIEEE Computer Society
Pages6765-6772
Number of pages8
ISBN (Electronic)9789887581536
DOIs
Publication statusPublished - 1 Jan 2022
Event41st Chinese Control Conference, CCC 2022 - Hefei, China
Duration: 25 Jul 202227 Jul 2022

Publication series

NameChinese Control Conference, CCC
Volume2022-July
ISSN (Print)1934-1768
ISSN (Electronic)2161-2927

Conference

Conference41st Chinese Control Conference, CCC 2022
Country/TerritoryChina
CityHefei
Period25/07/2227/07/22

Keywords

  • color equalization
  • image classification
  • peak signal-to-noise ratio
  • underwater color image quality evaluation

Fingerprint

Dive into the research topics of 'Underwater image enhancement and detection based on convolutional DCP and YOLOv5'. Together they form a unique fingerprint.

Cite this