matplotlib
numpy
opencv_python
Pillow
torch
torchvision
tqdm
