numpy
Pillow
tqdm
opencv-python
matplotlib
pandas
scikit-learn
segmentation-models-pytorch
torch-pruning==1.6.1
notebook
torch
torchvision
torchaudio
