h5py
scipy
Pillow
numpy
matplotlib
torch
torchvision