HG_paddle
July 12, 2023 · View on GitHub
Paper: A New Journey from SDRTV to HDRTV.
作者: Xiangyu Chen等
Paddle 复现版本
数据集
https://pan.baidu.com/s/1OSLVoBioyen-zjvLmhbe2g
提取码: 2me9
训练模型
链接:https://pan.baidu.com/s/1PxihzPH8WCgdRAtP9fivBg?pwd=hh66 提取码:hh66
训练步骤
train
#First
python train.py -opt options/train/train_pre_Gen.yml
#Then
python train.py -opt config/train/train_HG.yml
多卡仅需
```bash
python -m paddle.distributed.launch train.py --launcher fleet -opt config_file_path
测试步骤
python test.py -opt config/test/test_HG.yml
复现指标
| PSNR | |
|---|---|
| Paddle | 37.60 |
注:因显存限制,测试结果为测试图片降采样到1080p的结果