Please open up paint_to_numbers.py. The very last line in this script has all the power you need. Feel free to add an image to the image folder but I would recommend converting the size to be square ...