此超分辨率網絡使用的是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踐行者 的文章