requirements.txt

easydict==1.11
einops==0.8.1
gradio==4.0.0
kornia==0.6.3
matplotlib==3.5.0
numpy==1.24.4
opencv_python==4.6.0.66
opencv_python_headless==4.7.0.72
Pillow==8.4.0
pytorch_lightning==1.9.0
segmentation-models-pytorch@git+https://github.com/ragavsachdeva/segmentation_models.pytorch.git@0092ee4d6f851d89a4a401bb2dfa6187660b8dd3
torch==1.12.1
torchvision==0.13.1
tqdm==4.62.3