numpy>=1.18.5
torch>=1.4.0
torchvision>=0.5.0
sklearn
onnx==1.6.0
onnxruntime