Lightweight Deep Neural Network for Joint Learning of Underwater Object Detection and Color Conversion

Chia Hung Yeh, Chu Han Lin, Li Wei Kang*, Chih Hsiang Huang, Min Hui Lin, Chuan Yu Chang, Chua Chin Wang

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

97 Citations (Scopus)


Underwater image processing has been shown to exhibit significant potential for exploring underwater environments. It has been applied to a wide variety of fields, such as underwater terrain scanning and autonomous underwater vehicles (AUVs)-driven applications, such as image-based underwater object detection. However, underwater images often suffer from degeneration due to attenuation, color distortion, and noise from artificial lighting sources as well as the effects of possibly low-end optical imaging devices. Thus, object detection performance would be degraded accordingly. To tackle this problem, in this article, a lightweight deep underwater object detection network is proposed. The key is to present a deep model for jointly learning color conversion and object detection for underwater images. The image color conversion module aims at transforming color images to the corresponding grayscale images to solve the problem of underwater color absorption to enhance the object detection performance with lower computational complexity. The presented experimental results with our implementation on the Raspberry pi platform have justified the effectiveness of the proposed lightweight jointly learning model for underwater object detection compared with the state-of-the-art approaches.

Original languageEnglish
Pages (from-to)6129-6143
Number of pages15
JournalIEEE Transactions on Neural Networks and Learning Systems
Issue number11
Publication statusPublished - 2022 Nov 1


  • Convolutional neural networks
  • deep learning
  • lightweight deep model
  • underwater image processing
  • underwater object detection

ASJC Scopus subject areas

  • Software
  • Computer Science Applications
  • Computer Networks and Communications
  • Artificial Intelligence


Dive into the research topics of 'Lightweight Deep Neural Network for Joint Learning of Underwater Object Detection and Color Conversion'. Together they form a unique fingerprint.

Cite this