pyyaml==5.1
torch==1.7.1
torchvision==0.8.2
opencv-python-headless
scikit-image
kornia==0.5.11