This repository trains the Conditional GAN in both Pytorch and Tensorflow on the Fashion MNIST and Rock-Paper-Scissors dataset. It is tested with:
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
├── PyTorch
│ ├── CGAN-PyTorch.ipynb
│ └── cgan_pytorch.py
└── TensorFlow
├── CGAN-FashionMnist-TensorFlow.ipynb
├── cgan_fashionmnist_tensorflow.py
├── CGAN-RockPaperScissor-TensorFlow.ipynb
└── cgan_rockpaperscissor_tensorflow.py
To train the Conditional GAN with Pytorch, please go into the Pytorch
folder and execute the Jupyter Notebook.
To train the Conditional GAN with TensorFlow, please go into the Tensorflow
folder and execute the Jupyter Notebook.
Want to become an expert in AI? AI Courses by OpenCV is a great place to start.