A GAN-Based Input-Size Flexibility Model for Single Image Dehazing

by   Shichao Kan, et al.

Image-to-image translation based on generative adversarial network (GAN) has achieved state-of-the-art performance in various image restoration applications. Single image dehazing is a typical example, which aims to obtain the haze-free image of a haze one. This paper concentrates on the challenging task of single image dehazing. Based on the atmospheric scattering model, we design a novel model to directly generate the haze-free image. The main challenge of image dehazing is that the atmospheric scattering model has two parameters, i.e., transmission map and atmospheric light. When we estimate them respectively, the errors will be accumulated to compromise dehazing quality. Considering this reason and various image sizes, we propose a novel input-size flexibility conditional generative adversarial network (cGAN) for single image dehazing, which is input-size flexibility at both training and test stages for image-to-image translation with cGAN framework. We propose a simple and effective U-type residual network (UR-Net) to combine the generator and adopt the spatial pyramid pooling (SPP) to design the discriminator. Moreover, the model is trained with multi-loss function, in which the consistency loss is a novel designed loss in this paper. We finally build a multi-scale cGAN fusion model to realize state-of-the-art single image dehazing performance. The proposed models receive a haze image as input and directly output a haze-free one. Experimental results demonstrate the effectiveness and efficiency of the proposed models.


page 1

page 2

page 3

page 5

page 6

page 7

page 8

page 11


Efficient High-Resolution Image-to-Image Translation using Multi-Scale Gradient U-Net

Recently, Conditional Generative Adversarial Network (Conditional GAN) h...

Dehaze-GLCGAN: Unpaired Single Image De-hazing via Adversarial Training

Single image de-hazing is a challenging problem, and it is far from solv...

DeblurGAN-v2: Deblurring (Orders-of-Magnitude) Faster and Better

We present a new end-to-end generative adversarial network (GAN) for sin...

Single Image Haze Removal using a Generative Adversarial Network

Single image haze removal is an under constrained problem due to lack of...

Image Shape Manipulation from a Single Augmented Training Sample

In this paper, we present DeepSIM, a generative model for conditional im...

Single image dehazing for a variety of haze scenarios using back projected pyramid network

Learning to dehaze single hazy images, especially using a small training...

FD-GAN: Generative Adversarial Networks with Fusion-discriminator for Single Image Dehazing

Recently, convolutional neural networks (CNNs) have achieved great impro...

Please sign up or login with your details

Forgot password? Click here to reset