Skip to content

ABHISHEKgauti25/ImageColorization

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 

Repository files navigation

Photo Colorization Model

Welcome to the Photo Colorization Model repository! This project aims to provide a powerful machine-learning solution for transforming black-and-white photos into vibrant, realistic color images. The model uses advanced deep learning techniques and comprises an encoder, decoder, fusion layer, and classification layer.

Model Architecture

Our colorization model consists of the following components:

  1. Encoder: Extracts essential features from black and white images.
  2. Decoder: Generates high-resolution color images based on the encoded features.
  3. Fusion Layer: Combines encoded features with external references, enhancing colorization accuracy.
  4. Classification Layer: Understands and classifies different objects and scenes within the image, guiding the colorization process.

layer_including_fusion_layer

Acknowledgments

I would like to express our gratitude to the open-source community for their valuable contributions and to the developers of TensorFlow and other libraries utilized in this project. I would like to thank Emil Wallner for the amazing blog, which is the inspiration for this project. For the dataset, I would like to thank Aayush Sharma.

Contact

If you have any questions, suggestions, or feedback, please don't hesitate to reach out to me at Kaggle. We appreciate your interest in our project and look forward to hearing from you!

Happy colorizing!

Releases

No releases published

Packages

No packages published