albumentations==1.0.0
torch==1.9.0
torchvision==0.10.0

[test]
pytest
