opencv-python
torch
tqdm