matplotlib
numpy
torchvision
