torch
torchvision
pytorch-lightning
segmentation-models-pytorch
hydra-core
albumentations
pandas
tensorboardX
pillow
matplotlib
numpy
pytorch_toolbelt

[tests]
pytest
scikit-image
parameterized
