Skip to content

AntonioTepsich/Image-colorization-and-denoising

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image Colorization and Denoising with Pix2Pix and DnCNN

Authors

This repository was made by:

Adviser

Report in Spanish

You can view the full report here

Abstract

We developed and applied deep learning models to colorize black-and-white images, specifically using a DnCNN for noise reduction and a cGAN generator for colorization. Our study demonstrates that it is possible not only to convincingly restore colors but also to improve the visual quality of historical images, surpassing the limitations of their time. Despite certain limitations related to generalization in varied backgrounds and the quality of our own dataset images, the models showed significant ability to produce high-quality images.

Motivation

The intersection of memory and technological innovation drives our project, where we strive to keep precious moments alive. As technology has evolved, from canvases to photographs, many memories have been relegated to grayscale due to the limitations of the era, leaving the original colors forgotten. The goal of this work is to revive the colors of our family images, bringing them back to life with modern technology.

Results

Architecture

Model Architecture

Usage

1-Jupyter Notebook

Run the file: image_colorization.ipynb

2-Download the dataset from Kaggle

Choose which dataset to download in the second cell:

download_art_dataset()
download_celeba_dataset()

After executing, enter your Kaggle account credentials and API key:

Your Kaggle username: XXXX
Your Kaggle Key: XXXXXXXXXXXXXXXXXX

If you want to download a different dataset, search in the scripts folder.

About

with Pix2Pix and DnCNN

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •