This is the code for the paper AutoEnhancer: Transformer on U-Net Architecture search for Underwater Image Enhancement
The model link of the optimal network by the proposed methods is at https://drive.google.com/drive/folders/10dJI5Pb5NMYhCg0ffZKlTExut-_PQsVf?usp=sharing. And put the .pth file into the ckpt folder.
data folder is the testing images.
The train and test code are available. Please check and modify the image and GT path.
P.S: author is too lazy and doesn't want to write much comments.