pre-requirements.txt

torch==2.2.1
torchvision==0.17.1
torchmetrics==0.6.0
opencv-python==4.6.0.66
pytorch-lightning==1.4.9
omegaconf==2.3.0
einops==0.6.1
kornia==0.6.12