sketch-artist's Introduction

Sketch-Artist

Overview

A Tensorflow implementation of a Conditional GAN (Generative Adversarial Network) for generating human faces from a text description. This project was created for my CSC 340 (Artificial Intelligence) final project.

YouTube Video

TO-DO

Does not produce samples accurate to condition yet

Label Vector

The label is a vector with 5 indices each corresponding to a different facial feature

The values for each feature can either be 1 or -1

Black hair
Blonde hair
Brown hair
Male
Beard

Generator

Discriminator

Installation

Git clone the repository and cd into the directory

git clone https://github.com/greerviau/Sketch-Artist.git && cd Sketch-Artist

Download the CelebA dataset here and extract

In CGAN.py add data directory to CelebA object

Make sure that directory contains list_attr_celeba.csv and img_align_celeba

celebA = CelebA(output_size, channel, sample_size, batch_size, crop, data_dir=<path-to-data>)

Usage

python CGAN.py train
python CGAN.py test

Results

After 100 epochs

sketch-artist's People

Contributors

Stargazers

Watchers

sketch-artist's Issues

Incorrect Dataset URL

Dataset URL in readme.md is invalid (404) and needs needs to be updated to https://www.kaggle.com/jessicali9530/celeba-dataset/download

I tried running the code on Google Colab but faced the following errors:

Few issues I faced when I tried to run the code on Google Colab:
Issue 1:
Loading: 000001.jpg - Loaded: 0Traceback (most recent call last):
File "/content/Sketch-Artist/CGAN.py", line 346, in
cgan.train()
File "/content/Sketch-Artist/CGAN.py", line 92, in train
self.celebA.load_data()
File "/content/Sketch-Artist/utils.py", line 42, in load_data
h, w, c = image.shape
AttributeError: 'NoneType' object has no attribute 'shape'

Issue 2:
File "/content/Sketch-Artist/CGAN.py", line 346, in
cgan.train()
File "/content/Sketch-Artist/CGAN.py", line 83, in train
self.saver.restore(sess, self.model_dir+self.version+'/'+self.version+'.ckpt')
File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/training/saver.py", line 1282, in restore
checkpoint_prefix)
ValueError: The passed save_path is not a valid checkpoint: model/face_gen_per_batch_filtered_4/face_gen_per_batch_filtered_4.ckpt

Issue 3:
File "/content/Sketch-Artist/CGAN.py", line 346, in
cgan.train()
File "/content/Sketch-Artist/CGAN.py", line 117, in train
train_images, real_labels = self.celebA.get_next_batch(batch)
File "/content/Sketch-Artist/utils.py", line 88, in get_next_batch
length = len(self.data)
TypeError: object of type 'NoneType' has no len()

I am new to GANs. Kindly help me resolve these issues. Thanks

Recommend Projects

greerviau / sketch-artist Goto Github PK