├── .gitignore ├── LICENSE ├── README.md ├── assets ├── EXAMPLES.md ├── HiFIC_torch_colab_demo.ipynb ├── camp_jpg_compress.png ├── hific │ ├── CLIC2020_18_RECON_0.209bpp.png │ ├── CLIC2020_19_RECON_0.565bpp.png │ ├── CLIC2020_20_RECON_0.330bpp.png │ ├── CLIC2020_3_RECON_0.269bpp.png │ ├── CLIC2020_5_RECON_0.160bpp.png │ ├── camp_hific.png │ ├── cathedral_RECON_0.168bpp.png │ ├── clocktower_RECON_0.090bpp.png │ ├── satellite_RECON_0.039bpp.png │ └── telephone_RECON_0.083bpp.png └── originals │ ├── CLIC2020_18.png │ ├── CLIC2020_19.png │ ├── CLIC2020_20.png │ ├── CLIC2020_3.png │ ├── CLIC2020_5.png │ ├── camp_original.png │ ├── cathedral_13.2bpp.png │ ├── clocktower_9.93bpp.png │ ├── satellite_4.31bpp.png │ └── telephone_5.61bpp.png ├── compress.py ├── data ├── .gitignore └── originals │ └── .gitignore ├── default_config.py ├── experiments └── .gitignore ├── requirements.txt ├── src ├── README.md ├── __init__.py ├── compression │ ├── ans.py │ ├── compression_utils.py │ ├── entropy_coding.py │ ├── entropy_models.py │ ├── hyperprior_model.py │ └── prior_model.py ├── helpers │ ├── __init__.py │ ├── datasets.py │ ├── maths.py │ ├── metrics.py │ ├── pmf.cc │ └── utils.py ├── hyperprior.py ├── loss │ ├── __init__.py │ ├── losses.py │ └── perceptual_similarity │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── dist_model.py │ │ ├── networks_basic.py │ │ ├── perceptual_loss.py │ │ ├── pretrained_networks.py │ │ └── weights │ │ └── v0.1 │ │ ├── alex.pth │ │ ├── squeeze.pth │ │ └── vgg.pth ├── model.py ├── network │ ├── __init__.py │ ├── discriminator.py │ ├── encoder.py │ ├── generator.py │ └── hyper.py └── normalisation │ ├── __init__.py │ ├── channel.py │ └── instance.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/README.md -------------------------------------------------------------------------------- /assets/EXAMPLES.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/EXAMPLES.md -------------------------------------------------------------------------------- /assets/HiFIC_torch_colab_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/HiFIC_torch_colab_demo.ipynb -------------------------------------------------------------------------------- /assets/camp_jpg_compress.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/camp_jpg_compress.png -------------------------------------------------------------------------------- /assets/hific/CLIC2020_18_RECON_0.209bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/hific/CLIC2020_18_RECON_0.209bpp.png -------------------------------------------------------------------------------- /assets/hific/CLIC2020_19_RECON_0.565bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/hific/CLIC2020_19_RECON_0.565bpp.png -------------------------------------------------------------------------------- /assets/hific/CLIC2020_20_RECON_0.330bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/hific/CLIC2020_20_RECON_0.330bpp.png -------------------------------------------------------------------------------- /assets/hific/CLIC2020_3_RECON_0.269bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/hific/CLIC2020_3_RECON_0.269bpp.png -------------------------------------------------------------------------------- /assets/hific/CLIC2020_5_RECON_0.160bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/hific/CLIC2020_5_RECON_0.160bpp.png -------------------------------------------------------------------------------- /assets/hific/camp_hific.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/hific/camp_hific.png -------------------------------------------------------------------------------- /assets/hific/cathedral_RECON_0.168bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/hific/cathedral_RECON_0.168bpp.png -------------------------------------------------------------------------------- /assets/hific/clocktower_RECON_0.090bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/hific/clocktower_RECON_0.090bpp.png -------------------------------------------------------------------------------- /assets/hific/satellite_RECON_0.039bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/hific/satellite_RECON_0.039bpp.png -------------------------------------------------------------------------------- /assets/hific/telephone_RECON_0.083bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/hific/telephone_RECON_0.083bpp.png -------------------------------------------------------------------------------- /assets/originals/CLIC2020_18.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/originals/CLIC2020_18.png -------------------------------------------------------------------------------- /assets/originals/CLIC2020_19.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/originals/CLIC2020_19.png -------------------------------------------------------------------------------- /assets/originals/CLIC2020_20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/originals/CLIC2020_20.png -------------------------------------------------------------------------------- /assets/originals/CLIC2020_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/originals/CLIC2020_3.png -------------------------------------------------------------------------------- /assets/originals/CLIC2020_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/originals/CLIC2020_5.png -------------------------------------------------------------------------------- /assets/originals/camp_original.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/originals/camp_original.png -------------------------------------------------------------------------------- /assets/originals/cathedral_13.2bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/originals/cathedral_13.2bpp.png -------------------------------------------------------------------------------- /assets/originals/clocktower_9.93bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/originals/clocktower_9.93bpp.png -------------------------------------------------------------------------------- /assets/originals/satellite_4.31bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/originals/satellite_4.31bpp.png -------------------------------------------------------------------------------- /assets/originals/telephone_5.61bpp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/assets/originals/telephone_5.61bpp.png -------------------------------------------------------------------------------- /compress.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/compress.py -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore 3 | !reconstructions/ 4 | !originals/ -------------------------------------------------------------------------------- /data/originals/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/data/originals/.gitignore -------------------------------------------------------------------------------- /default_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/default_config.py -------------------------------------------------------------------------------- /experiments/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore 3 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/README.md -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/compression/ans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/compression/ans.py -------------------------------------------------------------------------------- /src/compression/compression_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/compression/compression_utils.py -------------------------------------------------------------------------------- /src/compression/entropy_coding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/compression/entropy_coding.py -------------------------------------------------------------------------------- /src/compression/entropy_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/compression/entropy_models.py -------------------------------------------------------------------------------- /src/compression/hyperprior_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/compression/hyperprior_model.py -------------------------------------------------------------------------------- /src/compression/prior_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/compression/prior_model.py -------------------------------------------------------------------------------- /src/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/helpers/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/helpers/datasets.py -------------------------------------------------------------------------------- /src/helpers/maths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/helpers/maths.py -------------------------------------------------------------------------------- /src/helpers/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/helpers/metrics.py -------------------------------------------------------------------------------- /src/helpers/pmf.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/helpers/pmf.cc -------------------------------------------------------------------------------- /src/helpers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/helpers/utils.py -------------------------------------------------------------------------------- /src/hyperprior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/hyperprior.py -------------------------------------------------------------------------------- /src/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/loss/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/loss/losses.py -------------------------------------------------------------------------------- /src/loss/perceptual_similarity/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/loss/perceptual_similarity/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/loss/perceptual_similarity/base_model.py -------------------------------------------------------------------------------- /src/loss/perceptual_similarity/dist_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/loss/perceptual_similarity/dist_model.py -------------------------------------------------------------------------------- /src/loss/perceptual_similarity/networks_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/loss/perceptual_similarity/networks_basic.py -------------------------------------------------------------------------------- /src/loss/perceptual_similarity/perceptual_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/loss/perceptual_similarity/perceptual_loss.py -------------------------------------------------------------------------------- /src/loss/perceptual_similarity/pretrained_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/loss/perceptual_similarity/pretrained_networks.py -------------------------------------------------------------------------------- /src/loss/perceptual_similarity/weights/v0.1/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/loss/perceptual_similarity/weights/v0.1/alex.pth -------------------------------------------------------------------------------- /src/loss/perceptual_similarity/weights/v0.1/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/loss/perceptual_similarity/weights/v0.1/squeeze.pth -------------------------------------------------------------------------------- /src/loss/perceptual_similarity/weights/v0.1/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/loss/perceptual_similarity/weights/v0.1/vgg.pth -------------------------------------------------------------------------------- /src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/model.py -------------------------------------------------------------------------------- /src/network/__init__.py: -------------------------------------------------------------------------------- 1 | # Model / loss definitions 2 | -------------------------------------------------------------------------------- /src/network/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/network/discriminator.py -------------------------------------------------------------------------------- /src/network/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/network/encoder.py -------------------------------------------------------------------------------- /src/network/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/network/generator.py -------------------------------------------------------------------------------- /src/network/hyper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/network/hyper.py -------------------------------------------------------------------------------- /src/normalisation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/normalisation/channel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/normalisation/channel.py -------------------------------------------------------------------------------- /src/normalisation/instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/src/normalisation/instance.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Justin-Tan/high-fidelity-generative-compression/HEAD/train.py --------------------------------------------------------------------------------