ANFIC-Image-Compression-Using-Augmented-Normalizing-Flows

ANFIC: Image Compression Using Augmented Normalizing Flows
Accepted by OJCAS'21
paper: https://arxiv.org/abs/2107.08470

Project Installation

Prepare PyTorch 1.4.0 environment and correspond torchvision
Run sh install.sh

How to run?

Checklist before running

Download pretrained weight from https://drive.google.com/drive/folders/1M2WaPFIGOJeBdDkH5Au3rMdanQP6zYIE?usp=sharing and unzip weights to ./models/

Evaluation

$ python ANFIC_codec.py -UC -C GaussianMixtureModel -NF 320 -VNFL 128 -QE -DM eval -ckpt ./models/ANFIC_R1.ckpt -SD ./example_image/ -TD ./recon/

Encode

$ python ANFIC_codec.py -UC -C GaussianMixtureModel -NF 320 -VNFL 128 -QE -DM compress -ckpt ./models/ANFIC_R1.ckpt -SD ./example_image/ --eval

Decode

$ python ANFIC_codec.py -UC -C GaussianMixtureModel -NF 320 -VNFL 128 -QE -DM decompress -ckpt ./models/ANFIC_R1.ckpt -SD ./example_image/ -TD ./recon/ --eval -OD ./example_image/

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
example_image		example_image
torchac		torchac
.gitignore		.gitignore
ANFIC_codec.py		ANFIC_codec.py
README.md		README.md
context_model.py		context_model.py
datasets.py		datasets.py
entropy_models.py		entropy_models.py
functional.py		functional.py
generalizedivisivenorm.py		generalizedivisivenorm.py
install.sh		install.sh
loss.py		loss.py
metric.py		metric.py
modules.py		modules.py
msssim.py		msssim.py
networks.py		networks.py
parameterizers.py		parameterizers.py
requirements.txt		requirements.txt
signalconv.py		signalconv.py
ssim.py		ssim.py
utils.py		utils.py

dororojames/ANFIC

Folders and files

Latest commit

History

Repository files navigation

ANFIC-Image-Compression-Using-Augmented-Normalizing-Flows

Project Installation

How to run?

Checklist before running

Evaluation

Encode

Decode

About

Resources

Stars

Watchers

Forks

Languages