albumentations
iglovikov_helper_functions
pytorch_lightning
pytorch_toolbelt
segmentation-models-pytorch
torch
tqdm
wandb

[test]
pytest
