torch>=1.7.0
torchvision>=0.8.1
opencv-python
numpy
tqdm
