Advertisement
f11894

EDSR-PyTorch 導入方法

Feb 10th, 2018
690
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.56 KB | None | 0 0
  1. 1 以下のサイトからAnaconda Python 3.6 versionをインストール
  2. https://www.anaconda.com/download/
  3.  
  4. 2 以下のサイトからGit for Windowsをインストール
  5. https://gitforwindows.org/
  6.  
  7. 3 コマンドプロンプトに以下のコマンドを入力してpytorchをインストール
  8. conda install pytorch-cpu -c pytorch -y
  9.  
  10. 4 コマンドプロンプトに以下のコマンドを入力してtorchvisionとtqdmをインストール
  11. pip install torchvision tqdm
  12.  
  13. 5 コマンドプロンプトに以下のコマンドを入力してEDSR-PyTorchをインストール
  14. git clone https://github.com/thstkdgus35/EDSR-PyTorch
  15.  
  16. 6 カレントディレクトリを変更
  17. cd EDSR-PyTorch\code
  18.  
  19. 7 以下のサイトからmodel_pytorch.tarをダウンロードして解凍し、モデルファイルをEDSR-PyTorchのexperiment/modelに配置
  20. https://github.com/thstkdgus35/EDSR-PyTorch#quick-start-demo
  21.  
  22. 8 EDSR-PyTorchのcode/trainer.pyの88行目の (lr, hr, filename, _) を (lr, hr, filename) に書き換えて保存する。
  23.  
  24. 9 コマンドプロンプトに以下ののコマンドを入力して超解像を開始する
  25.  
  26. EDSRで超解像
  27. python main.py --data_test Demo --cpu --scale 2 --n_threads 0 --n_resblocks 32 --n_feats 256 --res_scale 0.1 --chop --pre_train ../experiment/model/EDSR_x2.pt --test_only --save_results
  28.  
  29. MDSRで超解像
  30. python main.py --data_test Demo --cpu --scale 2+3+4 --n_threads 0 --n_resblocks 80 --chop --model MDSR --pre_train ../experiment/model/MDSR.pt --test_only --save_results
  31.  
  32. 備考
  33. メモリを非常に食うので注意
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement