Démarrage rapide avec YOLOv5 🚀#

Lance-toi dans le domaine dynamique de la détection d'objets en temps réel avec Ultralytics YOLOv5 ! Ce guide est conçu pour servir de point de départ complet aux passionnés d'IA et aux professionnels souhaitant maîtriser YOLOv5. De l'installation initiale aux techniques d'entraînement avancées, nous t'accompagnons. À la fin de ce guide, tu auras les connaissances nécessaires pour intégrer en toute confiance YOLOv5 dans tes projets en utilisant des méthodes de deep learning de pointe. Allumons les moteurs et envolons-nous vers YOLOv5 !

Installer#

Prépare le lancement en clonant le dépôt YOLOv5 et en configurant l'environnement. Cela garantit que tous les prérequis nécessaires sont installés. Vérifie que tu as Python>=3.8.0 et PyTorch>=1.8 prêts pour le décollage. Ces outils fondamentaux sont essentiels pour exécuter YOLOv5 efficacement.

git clone https://github.com/ultralytics/yolov5 # clone repository
cd yolov5
pip install -r requirements.txt # install dependencies

Inférence avec PyTorch Hub#

Découvre la simplicité de l'inférence PyTorch Hub avec YOLOv5, où les modèles sont téléchargés de manière transparente à partir de la dernière version de YOLOv5. Cette méthode exploite la puissance de PyTorch pour faciliter le chargement et l'exécution des modèles, rendant l'obtention de prédictions intuitive.

import torch

# Model loading
model = torch.hub.load("ultralytics/yolov5", "yolov5s")  # Can be 'yolov5n' - 'yolov5x6', or 'custom'

# Inference on images
img = "https://ultralytics.com/images/zidane.jpg"  # Can be a file, Path, PIL, OpenCV, numpy, or list of images

# Run inference
results = model(img)

# Display results
results.print()  # Other options: .show(), .save(), .crop(), .pandas(), etc. Explore these in the Predict mode documentation.

Inférence avec detect.py#

Exploite detect.py pour une inférence polyvalente sur diverses sources. Il récupère automatiquement les modèles de la dernière version de YOLOv5 et enregistre les résultats en toute simplicité. Ce script est idéal pour une utilisation en ligne de commande et pour intégrer YOLOv5 dans des systèmes plus vastes, prenant en charge des entrées telles que des images, des vidéos, des répertoires, des webcams et même des flux en direct.

python detect.py --weights yolov5s.pt --source 0                              # webcam
python detect.py --weights yolov5s.pt --source image.jpg                      # image
python detect.py --weights yolov5s.pt --source video.mp4                      # video
python detect.py --weights yolov5s.pt --source screen                         # screenshot
python detect.py --weights yolov5s.pt --source path/                          # directory
python detect.py --weights yolov5s.pt --source list.txt                       # list of images
python detect.py --weights yolov5s.pt --source list.streams                   # list of streams
python detect.py --weights yolov5s.pt --source 'path/*.jpg'                   # glob pattern
python detect.py --weights yolov5s.pt --source 'https://youtu.be/LNwODJXcvt4' # YouTube video
python detect.py --weights yolov5s.pt --source 'rtsp://example.com/media.mp4' # RTSP, RTMP, HTTP stream

Entraînement#

Reproduis les benchmarks du jeu de données COCO pour YOLOv5 en suivant les instructions d'entraînement ci-dessous. Les modèles et jeux de données nécessaires (comme coco128.yaml ou le coco.yaml complet) sont récupérés directement à partir de la dernière version de YOLOv5. L'entraînement de YOLOv5n/s/m/l/x sur un GPU V100 prend généralement 1/2/4/6/8 jours respectivement (note que les configurations d'entraînement multi-GPU fonctionnent plus rapidement). Maximise les performances en utilisant le --batch-size le plus élevé possible ou utilise --batch-size -1 pour la fonctionnalité AutoBatch de YOLOv5, qui trouve automatiquement la taille de batch optimale. Les tailles de batch suivantes sont idéales pour les GPU V100-16GB. Consulte notre guide de configuration pour plus de détails sur les fichiers de configuration de modèles (*.yaml).

# Train YOLOv5n on COCO128 for 3 epochs
python train.py --data coco128.yaml --epochs 3 --weights yolov5n.pt --batch-size 128

# Train YOLOv5s on COCO for 300 epochs
python train.py --data coco.yaml --epochs 300 --weights '' --cfg yolov5s.yaml --batch-size 64

# Train YOLOv5m on COCO for 300 epochs
python train.py --data coco.yaml --epochs 300 --weights '' --cfg yolov5m.yaml --batch-size 40

# Train YOLOv5l on COCO for 300 epochs
python train.py --data coco.yaml --epochs 300 --weights '' --cfg yolov5l.yaml --batch-size 24

# Train YOLOv5x on COCO for 300 epochs
python train.py --data coco.yaml --epochs 300 --weights '' --cfg yolov5x.yaml --batch-size 16

Pour conclure, YOLOv5 n'est pas seulement un outil de pointe pour la détection d'objets, c'est aussi le témoignage de la puissance du machine learning pour transformer notre façon d'interagir avec le monde grâce à la vision par ordinateur. Au fil de ta progression dans ce guide et de l'application de YOLOv5 à tes projets, rappelle-toi que tu es à l'avant-garde d'une révolution technologique, capable d'accomplir des exploits remarquables en computer vision. Si tu as besoin d'informations supplémentaires ou du soutien d'autres visionnaires, tu es invité sur notre dépôt GitHub, qui abrite une communauté florissante de développeurs et de chercheurs. Explore d'autres ressources telles que Ultralytics Platform pour la gestion des jeux de données et l'entraînement de modèles sans code, ou consulte notre page Solutions pour des applications concrètes et de l'inspiration. Continue d'explorer, continue d'innover et profite des merveilles de YOLOv5. Bonne détection ! 🌠🔍