项目作者: abhigyan13

项目描述 :
Image Colorization with DCGANS
高级语言: Jupyter Notebook
项目地址: git://github.com/abhigyan13/Image_colorization_with_GANS.git
创建时间: 2021-03-05T23:16:06Z
项目社区:https://github.com/abhigyan13/Image_colorization_with_GANS

开源协议:

下载


Image_colorization_with_GANS

Gray-scale Image Colorization of images using a Conditional Deep Convolutional Generative Adversarial Network (DCGAN). This is a PyTorch implementation of the Conditional DCGAN(Deep Convolutional Generative Adversarial Networks) based on the research paper Image Colorization using Generative Adversarial Networkshttps

The model is trained on a subset of Places-365 data containing 5000 images . I was not able to train on large dataset due to the non availability of GPU and other hardwares . The suggestions from the paper have been used for hyperparameter tuning . The model was trained for about 60 epochs on Google Colab . The model displays some good results on the training set which are shown below .

Drive Link for Dataset https://drive.google.com/drive/folders/1h_9OlPtBA1tHylqtBItxP9SncVgMM0Zm?usp=sharing

Network Architecture

march

Trained Losses

  1. Generator Loss Plot

gen

  1. Discriminator Loss Plot

disc

Some Results

—————- Gray Scale Image ———————————— Original Image——————————————- Generated Image ————
i5
i3
i16
i22 (1)
i20 (1)
i17