torch>=1.7
numpy
opencv-python-headless
setuptools