Following is the suggested way to install the dependencies of our code(The default numpy and pip installed by conda may need to be downgraded.):
conda create -n bdr_diffusion
conda activate bdr_diffusion
conda install pytorch==1.9.0 torchvision==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge
pip install tqdm fire einops ocnn scikit-image==0.18.2 scikit-learn==0.24.2 pytorch-lightning==1.6.1
Please refer to the scripts in scripts/ for the usage of our code.
bash scripts/train.sh
bash scripts/generate.sh
https://rec.ustc.edu.cn/share/b9e072b0-306a-11ef-91cb-dfb6325d6cc5