numpy
pillow
av
torch
torchvision
