IACC: Cross Illumination-Aware and Colour Correction for Underwater Images Under Mixed Natural and Artificial Lighting
python=3.9
pytorch=1.12.1
timm
ruamel.yaml
kornia- Download the UIEB dataset
- Divide the dataset into training set, validation set and test set
- edit train_path and valid_path in config.yaml
- activate your python environment
- python run.py
Test metrics from IQA-PyTorch