CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-20 04:42:19 +12:00
Code Issues Releases Wiki Activity

README: Add docker instruction

Browse source
This commit is contained in:
Atsushi Takayama 2016-09-02 01:23:55 +09:00
parent 281b95a014
commit c38353e264
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
README.md
View file

@ -251,3 +251,13 @@ th train.lua -model upconv_7 -model_dir models/my_model -method noise_scale -sca
th waifu2x.lua -model_dir models/my_model -m noise_scale -scale 2 -noise_level 1 -i images/miku_small.png -o output.png
```
You can check the performance of model with `models/my_model/noise1_scale2.0x_best.png`.
## Docker
Requires `nvidia-docker`.
```
docker build -t waifu2x .
nvidia-docker run -p 8812:8812 waifu2x th web.lua
nvidia-docker run -v `pwd`/images:/images waifu2x th waifu2x.lua -force_cudnn 1 -m scale -scale 2 -i /images/miku_small.png -o /images/output.png
```