numpy
torch
Pillow
