numpy
onnxruntime
matplotlib
torch>=1.8
torchvision
