torch
matplotlib
numpy
albumentations
opencv-python
pytorch_lightning
grad-cam