Pytorch Reimplementation for End-to-end Optimized Image Compression
CUDA_VISIBLE_DEVICES=0 python train.py --config examples/example/config.json -n baseline
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