torch>=1.7
torchvision
pyyaml
huggingface_hub
safetensors>=0.2
datasets
einops
tabulate
opencv-python
ttach
scikit-learn