pillow
torch
numpy
