torch
pyyaml
torchvision
opencv-python
