랭킹으로 돌아가기
Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.
vaeganpytorchtensorflowgenerative-modelmachine-learningrbmrestricted-boltzmann-machine
주요 지표
스타 성장
스타
7.5k
포크
2k
주간 성장
—
이슈
20
2k4k6k
2023년 1월2023년 8월2024년 3월2024년 10월2025년 5월2025년 12월2026년 7월
아티팩트PyPI
pip install generative-modelsREADME
Generative Models
Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow. Also present here are RBM and Helmholtz Machine.
Note:
Generated samples will be stored in GAN/{gan_model}/out (or VAE/{vae_model}/out, etc) directory during training.
What's in it?
Generative Adversarial Nets (GAN)
- Vanilla GAN
- Conditional GAN
- InfoGAN
- Wasserstein GAN
- Mode Regularized GAN
- Coupled GAN
- Auxiliary Classifier GAN
- Least Squares GAN
- Boundary Seeking GAN
- Energy Based GAN
- f-GAN
- Generative Adversarial Parallelization
- DiscoGAN
- Adversarial Feature Learning & Adversarially Learned Inference
- Boundary Equilibrium GAN
- Improved Training for Wasserstein GAN
- DualGAN
- MAGAN: Margin Adaptation for GAN
- Softmax GAN
- GibbsNet
Variational Autoencoder (VAE)
Restricted Boltzmann Machine (RBM)
Helmholtz Machine
Dependencies
- Install miniconda http://conda.pydata.org/miniconda.html
- Do
conda env create - Enter the env
source activate generative-models - Install Tensorflow
- Install Pytorch
관련 저장소