matplotlib
numpy
opencv-contrib-python
Pillow
torch
torchvision
