matplotlib
torchvision
numpy