Semi-Supervised image dehazing network

A semi-supervised image dehazing network was proposed which consists of the supervised branch and the unsupervised branch. In the supervision branch, the encoding-decoding neural network is used as the network structure, and the network is constrained by the supervision loss. In the unsupervised branch, two similar sub-networks are used to estimate the transmission map and atmospheric light, and the unsupervised loss is constructed through prior knowledge to constrain the unsupervised branch. In the semi-supervised image dehazing network, the supervised branch and the unsupervised branch will output dehazing result, respectively. Then by minimizing the reconstruction loss between the two images, the supervised and unsupervised branches are constrained to make the network more generalizable. The entire semi-supervised image dehazing network is trained in an end-to-end manner, and the supervised and unsupervised branch shares weights in the encoding part. Extensive experimental results show that the proposed method has good performance in image dehazing compared with six advanced dehazing methods.