numpy
Pillow
scikit-image
torch
torchvision
