torch>=1.6.0
torchvision
opencv-python
numpy
matplotlib
tqdm
