pyyaml==5.1
torch
torchvision
opencv-python==4.4.0.40
scikit-image
kornia==0.5.11