pillow
numpy
gradio
torchsummary
lightning
grad-cam
albumentations