torch>=1.10
torchvision
torchmetrics
numpy
PyYAML
easydict
opencv-python
Pillow
scikit-learn
scikit-image
imgaug
tabulate
kornia
matplotlib
timm==0.8.15.dev0
FrEIA
geomloss
pandas
grad_cam
pyzmq

[cuda]
faiss-gpu

[cuda11]
faiss-gpu-cu11

[cuda12]
faiss-gpu-cu12
