anananan116 / VQ-VAE-Image-Generation Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

Image generation using vector quantized autoencoders

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 27 Commits
autoencoders		autoencoders
configs		configs
configs_pixelsnail		configs_pixelsnail
data_utils		data_utils
prior_models		prior_models
results		results
resutls_pixelsnail		resutls_pixelsnail
trainers		trainers
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
extract_code.py		extract_code.py
generate_sample.ipynb		generate_sample.ipynb
run.py		run.py
run_pixelsnail.py		run_pixelsnail.py
sample.py		sample.py
visualization.ipynb		visualization.ipynb

Repository files navigation

VQ-VAE-VQ-GAN-Image-Generation

Image generation usinng vector quantized autoencoders

Download data from celebA-HQ here, and put unziped images into ./data_utils/raw_data

It needs about 40-50 GB of RAM(not GRAM) to load the data and transform them. I will implement transformation on the fly so it need less ram to prepare the data in future versions.