numpy
pillow
torch
