STSC

Semantic-aware Texture-Structure Feature Collaboration for Underwater Image Enhancement [IEEE ICRA 2022]

By Di Wang, Long Ma, Risheng Liu, and Xin Fan

Updates

[2022-05-23] Our paper is available online! [Paper]

Requirements

• CUDA 10.1
• Python 3.6 (or later)
• Pytorch 1.6.0
• Torchvision 0.7.0
• OpenCV 3.4

Get start

Please download the pretrained VGG model MyVGG.pt and vggfeature.pth and put them into the folder 'pretrain' Baidu Disk pretrained VGG model MyVGG.pt [code: c915] and a vggfeature.pth [code: 84zk]

1. Check the dataset path in train.py, and then run:

      python train.py

2. Check the model and image pathes in test_UIEB.py and then run:

      python test_UIEB.py

Please download the UIEB pretrained model and put it into folder './checkpoints/'

Baidu Disk UIEB pretrained model [code: mgey]

Dataset

Please download the following datasets:

• UIEB
• EUVP
• RUIE
• USOD

Experimental Results

Citation

@InProceedings{Wang_22_ICRA,
  author    = {Di Wang and Long Ma and Risheng Liu and Xin Fan},
  title     = {Semantic-aware Texture-Structure Feature Collaboration for Underwater
               Image Enhancement},
  booktitle = {{IEEE} International Conference on Robotics and Automation {ICRA}},
  pages     = {4592--4598},
  year      = {2022}
}