The key to achieve haze removal is to estimate a medium transmission map for an input hazy image.
Traditional methods to remove haze from images rely on estimating a transmission map.
Conventional patch-based haze removal algorithms (e. g. the Dark Channel prior) usually performs dehazing with a fixed patch size.
We train a generative adversarial network to learn the probability distribution of clear images conditioned on the haze-affected images using the Wasserstein loss function, using a gradient penalty to enforce the Lipschitz constraint.
In addition, to further enhance the performance of the method for haze removal, a patch-map-based DCP has been embedded into the network, and this module has been trained with the atmospheric light generator, patch map selection module, and refined module simultaneously.