RDN
该存储库是。
要求
PyTorch 1.0.0
脾气暴躁的1.15.4
枕头5.4.1
h5py 2.8.0
tqdm 4.30.0
火车
可以从下面的链接下载转换为HDF5的DIV2K,Set5数据集。
数据集
规模
类型
关联
DIV2K
2个
火车
DIV2K
3
火车
DIV2K
4
火车
第5集
2个
评估
第5集
3
评估
第5集
4
评估
否则,您可以使用prepare.py创建自定义数据集。
python train.py --train-file " BLAH_BLAH/DIV2K_x4.h5 " \
--eval-file " BLAH_BLAH/Set5_x4.h5 " \
--outputs-dir " BLAH_BLAH/outputs " \
1