numpy
tqdm
torch
av
opencv-python
torchvision
matplotlib
