numpy
scikit-learn
scikit-image
torch
torchvision
pytorch_lightning
matplotlib
albumentations