torch
torchvision
grad-cam
pillow
numpy
albumentations