# Pinned to the exact versions used to produce the published results
# (NVIDIA A10G, CUDA 13.0, Python 3.12). torch/torchvision built with +cu130;
# install the CUDA build matching your system from https://pytorch.org if needed.
torch==2.12.0
torchvision==0.27.0
timm==1.0.27
scikit-learn==1.9.0
numpy==2.4.6
pillow==12.2.0
matplotlib==3.10.9
pyyaml==6.0.3
trackio==0.26.0
huggingface_hub==1.17.0
# optional: only needed if you enable the 'clahe' augmentation ablation
opencv-python-headless>=4.8