Colour Variation Minimization Retinex Decomposition and Enhancement with a Multi-Branch Decomposition Network

Abstract

This paper proposes a colour variation minimization retinex decomposition and enhancement with a multi-branch decomposition network (CvmD-net) to remove single image darkness. The network overcomes the problem that retinex deep learning model relies on matching bright images to process dark images. Specifically, our method takes two stages to light up the darkness in initial images: image decomposition and brightness optimization. We propose an input constant feature prior mechanism (ICFP) based on reflection constant features. The mechanism extracts structure and colour from the input images and constrains the reflected images output from the decomposition model to reduce color distortion and artifacts. The noise amplification during decomposition is addressed by a multi-branch decomposition network. Sub-networks with different structures are employed to focus on different prediction tasks. This paper proposes a reference mechanism for input brightness. This mechanism optimizes the output brightness distribution by calculating the reference brightness of the dark images. Experimental results on two benchmark datasets, namely, LOL and ZeroDCE, demonstrate that the proposed method can better balance dense noise interference and colour restoration. For the evaluation on real images, we collect Skynet images at night to verify the performance of the proposed approach. Compared with the state-of-the-art non-reference retinex decomposition-enhancement models, this paper has the best brightness optimization.