CSRNet_paddle

July 11, 2023 · View on GitHub

Paper: Conditional Sequential Modulation for Efficient Global Image Retouching.

Paper

作者: Jingwen He等

Paddle 复现版本

数据集

非原数据集,对照为hdrtvnet论文数据

https://pan.baidu.com/s/1OSLVoBioyen-zjvLmhbe2g

提取码: 2me9

训练模型

最优权重:

链接:https://pan.baidu.com/s/1j5qy_cQscpvDpuzVNe9Y8A?pwd=hh66 提取码:hh66

训练步骤

train

python train.py -opt config/train/train_CSRNet.yml
多卡仅需
​```bash
python -m paddle.distributed.launch train.py --launcher fleet -opt config_file_path

测试步骤

python test.py -opt config/test/test_CSRNet.yml

复现指标

PSNR
Paddle37.36

注:因显存限制,测试结果为测试图片降采样到1080p的结果