神经网络
该存储库是PyTorch编写的 (SRCNN)的实现。
要求
PyTorch 1.0.0
脾气暴躁的1.15.4
枕头5.4.1
h5py 2.8.0
tqdm 4.30.0
火车
可以从下面的链接下载转换为HDF5的91图像Set5数据集。 提取代码:do34
否则,您可以使用prepare.py创建自定义数据集。
python train.py --train-file " path_to_train_file " \
--eval-file " path_to_eval_file " \
--outputs-dir " path_to_outputs_file " \
--scale 3 \
--lr 1e-4 \
1