torch>=1.11.0
numpy
torchvision>=0.12.0
kornia
omegaconf
matplotlib
pytest
