torch
torchvision
albumentations