Skip to content

In this project, we built a (simple) automatic image colorizer from scratch in PyTorch!

License

Notifications You must be signed in to change notification settings

priyathamhub/Auto-Image-Colorization

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Auto-Image-Colorization

In this project, we built a (simple) Automatic Image Colorizer from scratch in PyTorch !

In image colorization, our goal is to produce a colored image given a grayscale input image. This problem is challenging because it is multimodal -- a single grayscale image may correspond to many plausible colored images. As a result, traditional models often relied on significant user input alongside a grayscale image.

Recently, deep neural networks have shown remarkable success in automatic image colorization -- going from grayscale to color with no additional human input. This success may in part be due to their ability to capture and use semantic information (i.e. what the image actually is) in colorization, although we are not yet sure what exacly makes these types of models perform so well.

Dataset Link : http://data.csail.mit.edu/places/places205/testSetPlaces205_resize.tar.gz Download and unzip.

Output as below

res1

res2

Releases

No releases published

Packages

No packages published