torch
torchvision
pytorch_lightning
grad-cam
pillow
numpy