pillow
numpy
pytorch