albumentations
iglovikov_helper_functions
numpy
torch
torchvision

[test]
pytest
