├── LICENSE ├── README.md ├── assets ├── bitters.png └── wavevae.png ├── bitters ├── callbacks │ └── image_logger.py ├── loader.py ├── models │ └── vqvae.py ├── modules │ ├── discriminator │ │ └── model.py │ ├── image_tokenizer │ │ ├── dwt.py │ │ ├── tokenizer.py │ │ └── utils.py │ ├── losses │ │ ├── __init__.py │ │ ├── lpips.py │ │ └── vqperceptual.py │ └── vqvae │ │ ├── quantize.py │ │ └── vae.py └── utils │ └── util.py ├── configs ├── imagenet_wgan.yaml ├── imagenet_wnet.yaml └── imagenet_wvae.yaml ├── eval_vae.py ├── requirements.txt └── train_vae.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/README.md -------------------------------------------------------------------------------- /assets/bitters.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/assets/bitters.png -------------------------------------------------------------------------------- /assets/wavevae.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/assets/wavevae.png -------------------------------------------------------------------------------- /bitters/callbacks/image_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/callbacks/image_logger.py -------------------------------------------------------------------------------- /bitters/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/loader.py -------------------------------------------------------------------------------- /bitters/models/vqvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/models/vqvae.py -------------------------------------------------------------------------------- /bitters/modules/discriminator/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/modules/discriminator/model.py -------------------------------------------------------------------------------- /bitters/modules/image_tokenizer/dwt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/modules/image_tokenizer/dwt.py -------------------------------------------------------------------------------- /bitters/modules/image_tokenizer/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/modules/image_tokenizer/tokenizer.py -------------------------------------------------------------------------------- /bitters/modules/image_tokenizer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/modules/image_tokenizer/utils.py -------------------------------------------------------------------------------- /bitters/modules/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/modules/losses/__init__.py -------------------------------------------------------------------------------- /bitters/modules/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/modules/losses/lpips.py -------------------------------------------------------------------------------- /bitters/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /bitters/modules/vqvae/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/modules/vqvae/quantize.py -------------------------------------------------------------------------------- /bitters/modules/vqvae/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/modules/vqvae/vae.py -------------------------------------------------------------------------------- /bitters/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/bitters/utils/util.py -------------------------------------------------------------------------------- /configs/imagenet_wgan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/configs/imagenet_wgan.yaml -------------------------------------------------------------------------------- /configs/imagenet_wnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/configs/imagenet_wnet.yaml -------------------------------------------------------------------------------- /configs/imagenet_wvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/configs/imagenet_wvae.yaml -------------------------------------------------------------------------------- /eval_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/eval_vae.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MIMICLab/BITTERS/HEAD/train_vae.py --------------------------------------------------------------------------------