Pytorch Reimplementation for End-to-end Optimized Image Compression

Running Script

CUDA_VISIBLE_DEVICES=0 python train.py --config examples/example/config.json -n baseline

Data Preparation

We need to first prepare the training and validation data. The trainging data is from flicker.com. You can obtain the training data according to description of CompressionData

The training details is similar to our another repo compression