A single-image dehazing algorithm based on GAN network for color restoration in YUV color space is proposed(YUVRecolorDehazeNet, YUVRDNet). Inspired by the GAN image coloring task, the algorithm converts the hazy image to YUV color space, weakens the effect of haze in the Y channel, and performs color restoration on the image in the UV channel to achieve dehazing. Meanwhile, skip connections are used to fuse low-level and high-level features for more precise dehazing.
Pre-trained model: