opencv-python
numpy
pillow
torch
torchvision
plotly