torch
torchvision
lightning-bolts  
grad-cam  
pl_bolts  
pytorch_lightning  
torchmetrics  
numpy
pillow