ClassSR: A General Framework to Accelerate Super-Resolution Networks by Data Characteristic.
作者: Xiangtao Kong, Hengyuan Zhao, Yu Qiao, Chao Dong
Paddle 复现版本
分类之后训练集用于训练SR模块 https://aistudio.baidu.com/aistudio/datasetdetail/106261
脚本任务地址: https://aistudio.baidu.com/aistudio/clusterprojectdetail/2356381
链接:https://pan.baidu.com/s/1jf0UKI_wf7yRhwdA4AU5Kw 提取码:u9lr
python train.py -opt config/train/train_RCAN.yml
python train_ClassSR.py -opt config/train/train_ClassSR_RCAN.yml
多卡仅需
python -m paddle.distributed.launch train.py --launcher fleet -opt config_file_path
python -m paddle.distributed.launch train_ClassSR.py --launcher fleet -opt config_file_path
python test.py -opt config/test/test_RCAN.yml
python test_ClassSR.py -opt config/test/test_ClassSR_RCAN.yml
PSNR | FLOPs | Percent | |
---|---|---|---|
论文 | 26.39 | 21.22 | 65% |
Paddle | 26.35 | 21.37 | 65.5% |