requirements.txt

torch>=1.8.0
torchvision>=0.9.0
matplotlib==3.5.2
numpy>=1.18.5 # for gradio
opencv-python==4.5.5.64
opencv-python-headless==4.5.5.64
scipy==1.7.3
tqdm==4.64.0
Pillow==9.1.1
scikit-image==0.19.2
scikit-learn==1.0.2
scipy==1.7.3