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

Research output: Contribution to journalArticlepeer-review

Abstract

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
JournalIEEE Transactions on Neural Networks and Learning Systems
DOIs
Publication statusAccepted/In press - 2021

Keywords

  • Convolutional neural networks
  • deep learning
  • Deep learning
  • Distortion
  • Feature extraction
  • Image color analysis
  • lightweight deep model
  • Object detection
  • Task analysis
  • Training
  • underwater image processing
  • underwater object detection.

ASJC Scopus subject areas

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

Fingerprint

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