Final year project on Image Dehazing

Add the training datapath and testing datapath in train modeule and the path to where you want to save the model in test module. While using the pretrained model, input the path of the ".pt" file.