torch
torchvision
timm==0.8.15dev0
albumentations
opencv-python-headless
grad-cam