arenasys / pixelization_inference

Inference script for Pixelization

All credit to those guys, I just stripped down thier code to make it simple to use.

Usage

git clone https://github.com/arenatemp/pixelization_inference
pip install pillow torch torchvision numpy

Download the pretrained models into the pixelization_inference folder: pixelart_vgg19.pth alias_net.pth 160_net_G_A.pth

python pixelization.py --input input_file.png