numpy
scipy
opencv-python-headless
torch
torchvision
pyyaml
