albumentations
einops
grad-cam
gradio
scikit-image
timm
torch