File size: 859 Bytes
c9595c6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from huggingface_hub import hf_hub_download
import os
def download_yolo_model(repo_id="keizer77/samyolo2", filename="best.pt", output_dir="models"):
"""
Télécharge un modèle YOLOv5 depuis Hugging Face.
"""
os.makedirs(output_dir, exist_ok=True) # Créez le dossier si nécessaire
print(f"Téléchargement du modèle {filename} depuis le dépôt {repo_id}...")
model_path = hf_hub_download(repo_id=repo_id, filename=filename, cache_dir=output_dir)
print(f"Modèle téléchargé et sauvegardé dans : {model_path}")
return model_path
if __name__ == "__main__":
# Exemple d'utilisation
try:
downloaded_model_path = download_yolo_model()
print(f"Modèle prêt à être utilisé : {downloaded_model_path}")
except Exception as e:
print(f"Erreur lors du téléchargement du modèle : {e}")
|