kornia==0.6.2
matplotlib
pytest
pyyaml
randomname
torch>=1.11.0
torchvision>=0.12.0
tqdm
