torch>=1.12.0
torchvision>=0.13.0
timm>=0.6.0
Pillow>=8.0.0
numpy>=1.20.0

[all]
timm>=0.6.0
matplotlib>=3.4.0
grad-cam>=1.4.0

[dev]
pytest>=7.0.0
pytest-cov
black
isort
flake8
twine
build

[gradcam]
grad-cam>=1.4.0

[viz]
matplotlib>=3.4.0
