albumentations
iglovikov_helper_functions
pytorch_lightning
pytorch_toolbelt
segmentation-models-pytorch
torch
tqdm

[test]
pytest
