Skip to content

Use CNN to learn the alpha and beta parameter to adjust an image contrast and brightness.

Notifications You must be signed in to change notification settings

dht7166/Brightness_Contrast

Repository files navigation

Brightness_contrast

This project was one of my thanksgiving project, while I was waiting for my chicken to be done

This is a really simple one. This model strive to learn two parameters: alpha and beta. Every image is then transformed by I*alpha + beta, a simple linear transformation. These two transformations are simple linear transformations that could actually change the overall contrast and brightness of a pictures. Consider these two images, the left one is the low exposure image, and the second one is transformed using alpha = 2.019 and beta = 1.876 a1 a2

These two parameters are learned through the model. As you can see, given certain alpha and beta param, it could really change an image. In fact, this is actually the very standard and simple way to change the brightness of an image.

The model is a simple CNNs with MaxPool downsample. It is implemented in Keras. To train the model, simply edit training.py for appropriate epoch value and training data, and use it to train. The pre-trained weights is available in the github itself. I am also writing a webapp for this model.

About

Use CNN to learn the alpha and beta parameter to adjust an image contrast and brightness.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages