/ELiC-ReImplemetation

Implemetation of ELIC: Efficient Learned Image Compression with Unevenly Grouped Space-Channel Contextual Adaptive Coding

Primary LanguagePython

ELIC: Efficient Learned Image Compression withUnevenly Grouped Space-Channel Contextual Adaptive Coding.

A Pytorch Implementation of "ELIC: Efficient Learned Image Compression withUnevenly Grouped Space-Channel Contextual Adaptive Coding."

Note that This Is Not An Official Implementation Code.

More details can be found in the following paper:

@inproceedings{he2022elic,
  title={Elic: Efficient learned image compression with unevenly grouped space-channel contextual adaptive coding},
  author={He, Dailan and Yang, Ziming and Peng, Weikun and Ma, Rui and Qin, Hongwei and Wang, Yan},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={5718--5727},
  year={2022}
}

Related links

Available Checkpoint

lambda Link
0.45 0.45
0.15 0.15
0.032 0.032
0.016 0.0016
0.008 0.008
0.004 0.004

Training dataset

According to the paper, They train the models on the largest 8000 images picked from ImageNet dataset. so download the ImageNet

The preprocessing and selection of ImageNet dataset is same to QVRF:https://github.com/bytedance/QRAF.

Environment

This code is based on the CompressAI.

   pip3 install compressai==1.1.5
   pip3 install thop
   pip3 install ptflops
   pip3 install timm

Usage

Train Usage

cd Code
python3 train.py -d ./dataset --N 192 --M 320 -e 4000 -lr 1e-4 -n 8 --lambda 13e-3 --batch-size 16 --test-batch-size 16 --aux-learning-rate 1e-3 --patch-size 256 256 --cuda --save --seed 1926 --clip_max_norm 1.0

In ELIC, each model is finetune by 200 epoches.

python3 train.py -d ./dataset --N 192 --M 320 -e 4000 -lr 1e-4 -n 8 --lambda 13e-3 --batch-size 16 --test-batch-size 16 --aux-learning-rate 1e-3 --patch-size 256 256 --cuda --save --seed 1926 --clip_max_norm 1.0 --pretrained --checkpoint Pretrained4000epoch_checkpoint.pth.tar

Update the entropy model

python3 -m updata.py checkpoint -n updatacheckpoint-name

Test

python Inference.py --dataset ./dataset/Kodak --output_path ELIC_0450_ft_3980_Plateau -p ./ELIC_0450_ft_3980_Plateau.pth.tar --patch 64

RD Results

We trained the network and ask the RD points from the author.