pytest
tqdm
opencv-python-headless
numpy
torch
torchvision
