CSDN博客代码讲解地址:https://blog.csdn.net/weixin_52261094/article/details/128389448
- PyTorch 1.0.0
- Numpy 1.15.4
- Pillow 5.4.1
- h5py 2.8.0
- tqdm 4.30.0
Dataset | Scale | Type | Link |
---|---|---|---|
91-image | 2 | Train | Download |
91-image | 3 | Train | Download |
91-image | 4 | Train | Download |
Set5 | 2 | Eval | Download |
Set5 | 3 | Eval | Download |
Set5 | 4 | Eval | Download |
Download any one of 91-image and Set5 in the same Scale and then move them under ./datasets
as ./datasets/91-image_x2.h5
and ./datasets/Set5_x2.h5
--train-file "path_to_train_file" \
--eval-file "path_to_eval_file" \
--outputs-dir "path_to_outputs_file" \
--scale 3 \
--lr 1e-4 \
--batch-size 16 \
--num-epochs 400 \
--num-workers 0 \
--seed 123
将权重移动到该目录文件下: ./outputs
as ./outputs/x3/best.pth