albumentations
iglovikov_helper_functions
timm
torch

[test]
pytest
