Auto paint sketch based on CNN(from lineart to anime illustration)
This work utilizes about a million pairs of illustration and lineart, while the former is from Danbooru dataset and the latter is from lllyasviel's work. For now, the result is not satisfying and more efforts are needed. If you look for a better auto-paint network please have a look at this and this.
Open your shell or terminal:
git clone https://github.com/adamz799/Paints.git
cd V1
Then you need to download network to V1/ and run:
python app.py --lineart YOUR_LINEART_FILE_PATH --ref_img YOUR_REFERENCE_IMAGE_FILE_PATH
I work on this alone and have no time to implement a GUI, so all the interactive functions are achieved by OpenCV, which might be a little weird. In painting, you need to pick a color from a reference image and draw it to the lineart, then press 'r' in your keyboard to launch the network. It is fine to draw only a part of the lineart to see the result and then draw other parts to correct network's fault.
These pictures are from Danbooru dataset.
This lineart is searched from the Internet by search engine, if this violate you right please contact me or report an issue.
- CUDA
- CuDNN
- Python 3.5
- Numpy
- Pytorch 1.0.1
- OpenCV-python
This work takes following repo(s) as reference(s):