Repository to host customized AECR-Net model developed as part of 5th UG2+ challenge (CVPR 2022) Track 1.1
Contrastive Learning for Compact Single Image Dehazing, CVPR2021
We modify the official implementation of AECR-Net to use it's NH_train pretrained model to perform dehazing and provide output images which serves as the input for the object detection task downstream.
We fine tuned the NH_train pretrainedmodel by training it on a subset of the training set available for the competition and obtained the best model based on evaluation of the remaining subset of the training set.
The results on the finetuned model are not satisfactory when compared to DW-GAN as the latter was found to perform better for the dehazing task for the competition. It is unclear during the time of submission if tinkering with the finetuning would improve the performance.
Before finetuning: SSIM: 0.6288 PSNR:14.2729
After finetuning, with the best model: SSIM:0.8001 PSNR:20.0715
Input
Output
