torch
torchvision
numpy
opencv-python-headless
onnx
