DiffIR 代码复现

1、环境配置

conda create -n diffir python=3.9
conda activate diffir

去模糊环境：

sudo apt-get update
sudo apt install tmux
sudo apt install libgl1-mesa-glx
pip install basicsr 
pip install -r requirements.txt
pip install pandas 
sudo python3 setup.py develop
pip install einops 
pip install natsort 
pip uninstall nvidia_cublas_cu11

pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118

2、train

python DiffIR/train.py -opt options/train_DiffIRS1.yml

运行成功：

3、test

sh test.sh