vqvae
vqvae copied to clipboard
A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)
Hi, could you please add a license file (MIT license etc.) to this repository so we know what to do if we partially reused and referenced the codes? thanks!
Hi, Having trained the VQVAE model, I managed to extract the encoded code, i.e. `min_encoding_indices`. However, I found that its shape is (batch_size*image_size/16, 1). That is, take a batch of...
Could you tell me how to do image inpainting with vqvae?
Bumps [certifi](https://github.com/certifi/python-certifi) from 2019.3.9 to 2022.12.7. Commits 9e9e840 2022.12.07 b81bdb2 2022.09.24 939a28f 2022.09.14 aca828a 2022.06.15.2 de0eae1 Only use importlib.resources's new files() / Traversable API on Python ≥3.11 ... b8eb5e9 2022.06.15.1...
Bumps [pillow](https://github.com/python-pillow/Pillow) from 6.0.0 to 9.3.0. Release notes Sourced from pillow's releases. 9.3.0 https://pillow.readthedocs.io/en/stable/releasenotes/9.3.0.html Changes Initialize libtiff buffer when saving #6699 [@radarhere] Limit SAMPLESPERPIXEL to avoid runtime DOS #6700 [@wiredfool]...
Bumps [numpy](https://github.com/numpy/numpy) from 1.16.4 to 1.22.0. Release notes Sourced from numpy's releases. v1.22.0 NumPy 1.22.0 Release Notes NumPy 1.22.0 is a big release featuring the work of 153 contributors spread...
referencing this line: https://github.com/MishaLaskin/vqvae/blob/d761a999e2267766400dc646d82d3ac3657771d4/models/quantizer.py#L63-L64 From what I can read in the paper the loss is : ``` log p ... + ||sg[z_e(x)]-e||^2 + beta * ||z_e(x)-sg[e]||^2 ``` but in this...
Thanks for your awesome VQ-VAE implementation! I read the code [here](https://github.com/MishaLaskin/vqvae/blob/d761a999e2267766400dc646d82d3ac3657771d4/models/quantizer.py#L55-L60), which is for getting quantized vectors: https://github.com/MishaLaskin/vqvae/blob/d761a999e2267766400dc646d82d3ac3657771d4/models/quantizer.py#L55-L60 I am wondering why don't you just use the forward function of...