Skip to content

Files

Latest commit

b22c929 · Jan 18, 2023

History

History

Deep-Convolutional-GAN

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Jul 14, 2021
Jul 14, 2021
Jan 18, 2023
Jul 14, 2021

Deep Convolutional GAN in TensorFlow and PyTorch

This repository contains code for Deep Convolutional GAN in TensorFlow and PyTorch blogpost.

download

Package Dependencies

The repository also trains the Deep Convolutional GAN in both Pytorch and Tensorflow on Anime-Faces dataset. It is tested with the following CUDA versions.

  • Cuda-11.1
  • Cudnn-8.0

The Pytorch and Tensorflow scripts require numpy, tensorflow, torch. To get the versions of these packages you need for the program, use pip: (Make sure pip is upgraded. python3 -m pip install -U pip)

pip3 install -r requirements.txt 

Directory Structure

├── PyTorch
│   ├── DCGAN_Anime_Pytorch.ipynb
│   └── dcgan_anime_pytorch.py
└── TensorFlow
    ├── DCGAN_Anime_Tensorflow.ipynb
    └── dcgan_anime_tesnorflow.py

Instructions

PyTorch

To train the Deep Convolutional GAN with Pytorch, please go into the Pytorch folder and execute the Jupyter Notebook.

TensorFlow

To train the Deep Convolutional GAN with TensorFlow, please go into the Tensorflow folder and execute the Jupyter Notebook.

AI Courses by OpenCV

Want to become an expert in AI? AI Courses by OpenCV is a great place to start.

img