/Continual_Cross-domain_Image_Compression

Code for the paper "Continual Cross-domain Image Compression via Entropy Prior Guided Knowledge Distillation and Scalable Decoding"

Primary LanguagePython

Continual Cross-domain Image Compression via Entropy Prior Guided Knowledge Distillation and Scalable Decoding

[Uncompleted] Pytorch code for the paper "Continual Cross-domain Image Compression via Entropy Prior Guided Knowledge Distillation and Scalable Decoding".

TODO:

  • Datasets
  • Models
  • Train code
  • Evaluation code

This repository is based on CompressAI.

Installation

conda create -n ccic python=3.7
conda activate ccic
pip install compressai
pip install numpy==1.19.1
pip install tqdm

Usage

Training

CUDA_VISIBLE_DEVICES='1' python train.py --model mbt2018 --filters 128 192 --lambda 0.032 --checkpoints_dir /path/to/checkpoint/folder/ --epochs 10000 10000 10000 10000 10000 --save_interval 200 -lr 1e-4

Datasets