Official Pytorch implementation for Deep Contextual Video Compression, NeurIPS 2021

Official Pytorch Implementation for Deep Contextual Video Compression, NeurIPS 2021

• Python 3.8 and conda, get Conda
• CUDA 11.0
• Environment

  conda create -n $YOUR_PY38_ENV_NAME python=3.8
  conda activate $YOUR_PY38_ENV_NAME
  
  pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
  python -m pip install -r requirements.txt
  

Currenlty the spatial resolution of video needs to be cropped into the integral times of 64.

The dataset format can be seen in dataset_config_example.json.

For example, one video of HEVC Class B can be prepared as:

• Crop the original YUV via ffmpeg:

   

  To finish reading, please visit source site