Computer and Modernization ›› 2024, Vol. 0 ›› Issue (10): 55-60.doi: 10.3969/j.issn.1006-2475.2024.10.009

Previous Articles     Next Articles

Domain Adaption-based Underwater Image Enhancement Algorithm

  

  1. (School of Electronic Information and Artificial Intelligence, Shaanxi University of Science & Technology, Xi’an 710021, China)
  • Online:2024-10-29 Published:2024-10-30

Abstract: Underwater image enhancement is a key technology for underwater missions. Aiming at the problems of color distortion and image blurring in underwater images, this paper designs an underwater domain adaptation network (UDA Net) based on the domain adaptive method to achieve effective enhancement of underwater images under unsupervised conditions. The sharpness of the original underwater images is significantly improved. Based on U-Net network framework, the algorithm uses convolutional neural network and multi-head attention mechanism for feature extraction, introduces adversarial learning idea, and adds discrimination network to domain feature extraction module and output module. Meanwhile, it optimizes feature enhancement loss, feature alignment loss and output alignment loss in the source domain to ensure style transfer and feature alignment from the source domain to the target domain, achieving underwater enhancement. In addition, the public underwater data sets EUVP, UIEB and UFO-120 are used for experimental verification, and the experimental results are compared with cutting-edge enhancement algorithms. The effectiveness of UDA Net algorithm is proved, and it has a good application prospect in underwater image enhancement tasks.

Key words:  ,  , underwater image enhancement; style transfer; adversarial learning; loss function

CLC Number: