Updates to the repository will be made available soon.

Installations

cuda-11.0
cmake-3.7.2-gcc-6.2.0
pytorch==1.7.1 
torchvision==0.8.2 
torchaudio==0.7.2 
numpy==1.20.3 
open3d==0.9.0.0 
einops==0.3.2 
scikit-learn==1.0.1 
compressai 
argparse
tensorboard

Training

./scripts/train.sh

Testing

./scripts/test_eval_exp.sh