Semantic-aware Texture-Structure Feature Collaboration for Underwater Image Enhancement [IEEE ICRA 2022]
By Di Wang, Long Ma, Risheng Liu, and Xin Fan
[2022-05-23] Our paper is available online! [Paper]
- CUDA 10.1
- Python 3.6 (or later)
- Pytorch 1.6.0
- Torchvision 0.7.0
- OpenCV 3.4
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]
- Check the dataset path in train.py, and then run:
python train.py
- 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]
Please download the following datasets:
@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}
}