此超分辨率网络使用的是Flickr1024数据集.
代码已经基于python=3.6, cuda=9.0做了测试。
依赖:
pytorch (0.4),
torchvision (0.2)
Matlab (For training/test data generation)
准备训练数据:
下载Flickr1024数据集,把图像放到data/train/Flickr1024目录下。
然后运行enerate_trainset.m,产生训练数据。
训练:
python train.py --scale_factor 4 --device cuda:0 --batch_size 32 --n_epochs 80 --n_steps 30
演示:
python demo_test.py --scale_factor 4 --device cuda:0 --dataset KITTI2012
閱讀更多 AI踐行者 的文章