torch
torchvision
pillow
fpsample
pytorch-lightning
numpy
tqdm