IR_SDE 代码复现

1、环境配置

conda create -n irsde python=3.9

conda activate irsde

cd IR_SDE
pip install -r requirements.txt

pip install tensorboard

pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu118

pip install ema_pytorch

2、train

进入目录修改数据集路径

cd IR_SDE/codes/config/deraining/options/train/ir-sde.yml

训练：

cd codes/config/deraining

python3 train.py -opt=options/train/ir-sde.yml

3、test

python test.py -opt=options/test/ir-sde.yml