/GAN-pytorch

Generating images using GAN w/ Pytorch

Primary LanguagePython

Welcome to GAN-Pytorch 👋

Version Documentation Maintenance

Generating Images Using GAN (Dogs, Cats..) w/ Pytorch

🏠 Homepage

✨ Description

Data

Store images as data/{--object}/imageSequence/00001.png
object is an argument that the GAN Network will generate.
List of Objects = [airplane, car, cat , dog, flower, fruit, motorbike, person] / Default : dog

Prerequisites

  • python >= 3.7
  • pip >= 20.1.1
  • pytorch >= 1.6.0
  • torchvision >= 0.7.0

Install

install -r requirements.txt

Usage

python main.py

Author

👤 Yoonseo Kim

🤝 Contributing

Contributions, issues and feature requests are welcome!
Feel free to check issues page.

Show your support

Give a ⭐️ if this project helped you!

References

  1. Dataset (https://www.kaggle.com/prasunroy/natural-images)
  2. Paper (https://arxiv.org/pdf/1406.2661.pdf)
  3. Basecode (https://github.com/eriklindernoren/PyTorch-GAN)