From 20d0f70d2abd01350d43aed47b4adee1d80df28d Mon Sep 17 00:00:00 2001 From: nagadomi Date: Sun, 17 May 2015 15:18:33 +0900 Subject: [PATCH] add README --- README.md | 111 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 111 insertions(+) create mode 100644 README.md diff --git a/README.md b/README.md new file mode 100644 index 0000000..0233399 --- /dev/null +++ b/README.md @@ -0,0 +1,111 @@ +# waifu2x + +Image Super-Resolution for anime/fan-art using Deep Convolutional Neural Networks. + +Demo-Application can be found at http://waifu2x.udp.jp/ . + +## Summary + +Click to see the slide show. + +![slide](https://raw.githubusercontent.com/nagadomi/waifu2x/master/images/slide.png) + +## References + +waifu2x is inspired by SRCNN [1]. 2D character picture (HatsuneMiku) is licensed under CC BY-NC by piapro [2]. + +- [1] Chao Dong, Chen Change Loy, Kaiming He, Xiaoou Tang, "Image Super-Resolution Using Deep Convolutional Networks", http://arxiv.org/abs/1501.00092 +- [2] "For Creators", http://piapro.net/en_for_creators.html + +## Dependencies + +### Platform +- [Torch7](http://torch.ch/) +- [NVIDIA CUDA](https://developer.nvidia.com/cuda-toolkit) +- [NVIDIA cuDNN](https://developer.nvidia.com/cuDNN) + +### Packages (luarocks) +- [cudnn](https://github.com/soumith/cudnn.torch) +- [graphicsmagick](https://github.com/clementfarabet/graphicsmagick) +- [turbo](https://github.com/kernelsauce/turbo) +- md5 +- uuid + +NOTE: Turbo 1.1.3 has bug in file uploading. Please install from the master branch on github. + +## Running The Web Application + +Please edit the first line in `web.lua`. +``` +local ROOT = '/path/to/waifu2x/dir' +``` +Run. +``` +th web.lua +``` + +View at: http://localhost:8812/ + +## Command line tools + +### Noise Reduction +``` +th waifu2x.lua -m noise -noise_level 1 -i input_image.png -o output_image.png +``` +``` +th waifu2x.lua -m noise -noise_level 2 -i input_image.png -o output_image.png +``` + +### 2x Upscaling +``` +th waifu2x.lua -m scale -i input_image.png -o output_image.png +``` + +### Noise Reduction + 2x Upscaling +``` +th waifu2x.lua -m noise_scale -noise_level 1 -i input_image.png -o output_image.png +``` +``` +th waifu2x.lua -m noise_scale -noise_level 2 -i input_image.png -o output_image.png +``` + +See also `images/gen.sh`. + +## Training Your Own Model + +### Data Preparation + +Genrating a file list. +``` +find /path/to/image/dir -name "*.png" > data/image_list.txt +``` +(You should use PNG! In my case, waifu2x is trained by 3000 PNG images.) + +Converting training data. +``` +th convert_data.lua +``` + +### Training a Noise Reduction(level1) model + +``` +th train.lua -method noise -noise_level 1 -test images/miku_noise.png +th cleanup_model.lua -model models/noise1_model.t7 -oformat ascii +``` +You can check the performance of model with `models/noise1_best.png`. + +### Training a Noise Reduction(level2) model + +``` +th train.lua -method noise -noise_level 2 -test images/miku_noise.png +th cleanup_model.lua -model models/noise2_model.t7 -oformat ascii +``` +You can check the performance of model with `models/noise2_best.png`. + +### Training a 2x Upsclaing model + +``` +th train.lua -method scale -scale 2 -test images/miku_small.png +th cleanup_model.lua -model models/scale2.0x_model.t7 -oformat ascii +``` +You can check the performance of model with `models/scale2.0x_best.png`.