football-minimap-generator / requirements.txt
RamziBm's picture
feat: Intégration de la détection de ballon avec YOLO et amélioration des vérifications de modèles dans app.py et main.py.
41f1119
raw
history blame contribute delete
736 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>=1.8.0
torchvision
torchaudio
numpy
opencv-python
pytorch-lightning>=1.4.0
soccernet
kornia>=0.6.0
# Ajouté car requis par sn_segmentation
# Dépendances pour l'estimation de pose (ViTPose)
transformers>=4.0.0
supervision
Pillow
accelerate
# scikit-learn # Retiré car K-Means n'est plus utilisé
gradio
# Ball detection
ultralytics
# Optionnel mais recommandé pour la gestion des chemins
pathlib
# Dépendances possibles de TvCalib (à vérifier)
# tensorboard
# matplotlib
# scipy
# tqdm
# scikit-image