torch>=1.4.0
numpy
pybind11
pycocotools
