albumentations
iglovikov_helper_functions
segmentation-models-pytorch
torch

[test]
pytest
