Tim M b22c929a64 Corrected additional `.md` files with new image		2 gadi atpakaļ
..
PyTorch	fe8fa36d17 CGAN Codebase	3 gadi atpakaļ
TensorFlow	fe8fa36d17 CGAN Codebase	3 gadi atpakaļ
readme.md	b22c929a64 Corrected additional `.md` files with new image	2 gadi atpakaļ
requirements.txt	ca2c1ac9cc CGAN Codebase	3 gadi atpakaļ

Conditional GAN in TensorFlow and PyTorch

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

Package Dependencies

This repository also trains the Conditional GAN in both Pytorch and Tensorflow on the Fashion MNIST and Rock-Paper-Scissors 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
│   ├── CGAN-PyTorch.ipynb
│   └── cgan_pytorch.py
└── TensorFlow
    ├── CGAN-FashionMnist-TensorFlow.ipynb
    ├── cgan_fashionmnist_tensorflow.py
    ├── CGAN-RockPaperScissor-TensorFlow.ipynb
    └── cgan_rockpaperscissor_tensorflow.py

Instructions

PyTorch

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

TensorFlow

To train the Conditional 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.

readme.md