torch>=1.9.1
torchvision>=0.3
numpy
opencv-python
matplotlib
kornia>=0.6.11