GRL CVPR2023图像修复 使用笔记
目录
github地址：
训练自己的数据教程：
网络学习：
损失函数：
github地址：
GitHub - ofsoundof/GRL-Image-Restoration
预训练模型，在release里面
如何使用代码：
conda create -n LightningIR python=3.8conda activate LightningIRpip install -r requirements.txt- 准备数据集
- 下载预训练模型
-
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