pillow
numpy
gradio
torchsummary
pytorch-lightning
grad-cam
torch-lr-finder
torch
torchvision
numpy
albumentations
lightning