opencv-python-headless
pyyaml==5.1
torch
torchvision