Home

Awesome

TCTL-Net: Template-free Color Transfer Learning for Self-Attention Driven Underwater Image Enhancement

This repository is the official PyTorch implementation of TCTL-Net.

Dataset preparation

Train

python train.py --dataroot /path_to_data --name train_name

Test

python test.py --dataroot /path_to_data --name test_name

You can download the trained model from Baidu Netdisk(Extraction code:rw85) or Google Drive.