Pillow
torch
numpy
