torch
torchvision
torch-lr-finder
grad-cam
pillow
numpy
pytorch_lightning
albumentations
matplotlib
opencv-python