torch
torchvision
# torch-lr-finder
grad-cam
pillow
numpy
opencv-python
# pytorch-lightning
# albumentations