Skip to content

Gray-scale to colored image using auto-encoder architecture. This project was made for hackathon conducted by iitg.ai club, IIT Guwahati

Notifications You must be signed in to change notification settings

haxer-max/Image_Colorization

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Image_Colorization

Why was this made

This project was made for hacathon conducted by IITG.AI club, IIT Guwahati.

I achieved rank 2 in this hackathon.

How does it work

This AI works on basis of Autoencoders. It works using bottleneck architecture. First the Convolutional Layers works as encoders and Transposed Convolutional Layers decodes the image.

It converts gray scale image to LAB color space.

It also uses skip connections from encoder layers to decoder layers to avoid the problem of vanishing gradients and so that gradient can backpropogate faster.

Model Details

totals params= 100k Loss function: MSE Optimiser: RMSprop output activation: Sigmoid

Examples

Input output

About

Gray-scale to colored image using auto-encoder architecture. This project was made for hackathon conducted by iitg.ai club, IIT Guwahati

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published