numpy
torch
Pillow