Skip to content

A simple Convolutional Neural Network (CNN) that learns to colorize grayscale images using the CIFAR-10 dataset. Built with TensorFlow and Keras, this project demonstrates basic image preprocessing, CNN design, and image reconstruction from single-channel (grayscale) input to three-channel (RGB) output.

Notifications You must be signed in to change notification settings

HDJadeja/gray2color-cnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

3 Commits
Β 
Β 
Β 
Β 

Repository files navigation

Gray2Color-CNN

This project builds a Convolutional Neural Network (CNN) to automatically colorize grayscale images using the CIFAR-10 dataset.

Features

  • Converts grayscale 32x32 images into full-color RGB images
  • Uses TensorFlow and Keras for model building and training
  • Simple, fully convolutional network design
  • Trained on CIFAR-10 sample images

Future Improvements

  • Train on larger datasets like CelebA or ImageNet
  • Improve loss function (e.g., perceptual loss)

πŸ“š Technologies Used

  • Python 3
  • TensorFlow
  • Matplotlib (for visualizations)

sample output

download download download

About

A simple Convolutional Neural Network (CNN) that learns to colorize grayscale images using the CIFAR-10 dataset. Built with TensorFlow and Keras, this project demonstrates basic image preprocessing, CNN design, and image reconstruction from single-channel (grayscale) input to three-channel (RGB) output.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published