grad-cam
gradio
torch
torchvision
torch-lr-finder
pillow
numpy
pytorch_lightning
torchmetrics
albumentations