opencv-python-headless
pandas
numpy
matplotlib
torch
torchvision