To implement our code, a pytorch3d environment should be installed:
conda env create -f environment.yml
conda activate pytorch3d
conda install pytorch3d=0.2.5 -c pytorch3d -y
'Main.py': optimiation code
To optimize noised data:
CUDA_VISIBLE_DEVICES=0 sh run.sh