目录

github地址：

训练自己的数据教程：

网络学习：

损失函数：

---

github地址：

GitHub - ofsoundof/GRL-Image-Restoration

预训练模型，在release里面

如何使用代码：

1. conda create -n LightningIR python=3.8
2. conda activate LightningIR
3. pip install -r requirements.txt
4. 准备数据集
5. 下载预训练模型

6.  torchx run -- -j 1x2 -- \-m training=False gpus=2 experiment=dm/grl model=grl/grl_small \load_state_dict=True pretrained_checkpoint="${MODEL_ZOO}/GRL/dm_grl_small.ckpt"

   训练自己的数据教程：

   GRL(CVPR2023图像修复)训练自己的数据集_grl image restoration如何测试-CSDN博客

网络学习：

损失函数：

losses/losses.py 

GANLoss

PerceptualLoss

CharbonnierLoss

TVLoss