Skip to content

LouisYuxuLu/ERANet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

35 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Real-Time Multi-Scene Visibility Enhancement for Promoting Navigational Safety of Vessels Under Complex Weather Conditions


Real-Time Multi-Scene Visibility Enhancement for Promoting Navigational Safety of Vessels Under Complex Weather Conditions
Ryan Wen Liu, Yuxu Lu*, Yuan Gao, Wenqi Ren, Fenghua Zhu, and Fei-Yue Wang (* indicates corresponding author)

Abstract: The visible-light camera, which is capable of environment perception and navigation assistance, has emerged as an essential imaging sensor for marine surface vessels in intelligent waterborne transportation systems (IWTS). However, the visual imaging quality inevitably suffers from several kinds of degradations (e.g., limited visibility, low contrast, color distortion, etc.) under complex weather conditions (e.g., haze, rain, and low-lightness). The degraded visual information will accordingly result in inaccurate environment perception and delayed operations for navigational risk. To promote the navigational safety of vessels, many computational methods have been presented to perform visual quality enhancement under poor weather conditions. However, most of these methods are essentially specific-purpose implementation strategies, only available for one specific weather type. To overcome this limitation, we propose to develop a general-purpose multi-scene visibility enhancement method, i.e., edge reparameterization- and attention-guided neural network (ERANet), to adaptively restore the degraded images captured under different weather conditions. In particular, our ERANet simultaneously exploits the channel attention, spatial attention, and reparameterization technology to enhance the visual quality while maintaining low computational cost. Extensive experiments conducted on standard and IWTS-related datasets have demonstrated that our ERANet could outperform several representative visibility enhancement methods in terms of both imaging quality and computational efficiency. The superior performance of IWTS-related object detection and scene segmentation could also be steadily obtained after ERANet-based visibility enhancement under complex weather conditions.


News 🚀

  • 2024.09.02: Paper is released on ArXiv.
  • 2024.09.02: ERANet is accepted by [IEEE TITS].

Requirement

  • Python 3.7
  • Pytorch 1.12.0

Motivation

Image

Network Architecture

Image

Test

  • Place the pre-training weight in the checkpoint folder.
  • For the dehazing task, place test hazy images in the input/hazy folder. Modify the Type=2, the line 54 of test.py.
  • For the deraining task, place test rainy images in the input/rainy folder. Modify the Type=0, the line 54 of test.py.
  • For the low-light enhancement task, place test low-light images in the input/low folder. Modify the Type=1, the line 55 of test.py.
  • Run test.py
  • The results are saved in output/hazy folder (dehazing), output/rainy folder (deraining), output/low folder (low-light enhancement).

Dehazing Results

Image

Deraining Results

Image

Low-light Enhancement Results

Image

Citation

@article{liu2024real,
  title={Real-Time Multi-Scene Visibility Enhancement for Promoting Navigational Safety of Vessels Under Complex Weather Conditions},
  author={Liu, Ryan Wen and Lu, Yuxu and Gao, Yuan and Guo, Yu and Ren, Wenqi and Zhu, Fenghua and Wang, Fei-Yue},
  journal={arXiv preprint arXiv:2409.01500},
  year={2024}
}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages