football-minimap-generator / requirements.txt
RamziBm's picture
init
bdb955e
raw
history blame
578 Bytes
# Installez PyTorch en utilisant la commande spécifique si nécessaire :
# pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu121
# Dépendances principales
torch
torchvision
torchaudio
numpy
opencv-python
pytorch-lightning
soccernet
kornia
# Ajouté car requis par sn_segmentation
# Dépendances pour l'estimation de pose (ViTPose)
transformers
supervision
Pillow # Souvent une dépendance de transformers/supervision, mais explicite ici
accelerate
# scikit-learn # Retiré car K-Means n'est plus utilisé
gradio