torch
torchvision
lightning-bolts  
grad-cam  
pytorch_lightning  
torchmetrics  
numpy
pillow
seaborn
pandas