Forráskód Böngészése

README: Add docker instruction

Atsushi Takayama 8 éve
szülő
commit
c38353e264
1 módosított fájl, 10 hozzáadás és 0 törlés
  1. 10 0
      README.md

+ 10 - 0
README.md

@@ -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
+```