requirements.txt

einops
importlib-metadata
mat4py
matplotlib
modelindex
numpy<2
rich
coverage
interrogate
pytest
albumentations>=0.3.2 --no-binary qudida,albumentations    # For Albumentations data transform
grad-cam >= 1.3.7,<1.5.0   # For CAM visualization
requests            # For torchserve
scikit-learn        # For t-SNE visualization and unit tests.