torch
torchvision
torchvision.transforms as transforms
torch.utils.data import DataLoader
numpy
scipy