Inférence locale avec YOLOv5 sur une webcam
Clonez le référentiel Ultralytics YOLOv5 Github sur votre ordinateur et exécutez detect.py après avoir enregistré les pondérations du modèle à partir de votre entraînement de modèle YOLOv5 personnalisé .
- Notez le répertoire (dossier) dans lequel vous clonez le référentiel. Enregistrez le chemin du fichier, si vous le pouvez, pour un accès facile plus tard.
- Aide-mémoire des commandes GitHub
Incluez les poids du modèle dans le dossier "yolov5" qui a été cloné sur votre système de fichiers (ordinateur) à partir de GitHub.
Exécution du modèle de webcam
Une fois que les poids personnalisés ont été ajoutés au répertoire yolov5 sur votre ordinateur local, vous pouvez exécuter le modèle dans votre webcam après être passé au répertoire yolov5 dans votre terminal.
Commande de base : exécuter YOLOv5 sur une webcam
Comme le répertoire "yolov5" dans l'exemple d'image réside dans le répertoire "mtraor3", et que nous utilisons les poids du modèle yolov5s enregistrés dans le dossier yolov5 ( /Users/mtraor3/yolov5 ) , l'exemple de code que nous entrons dans notre terminal pour exécuter localement sur la webcam ressemblerait à ceci :
python ./Users/mtraor3/yolov5/detect.py --weights ./Users/mtraor3/yolov5/yolov5s.pt --save_text --save_conf --conf-thres 0.40 --source 0
Enregistrement des prédictions et des scores de confiance dans un fichier .txt avec inférence effectuée à un seuil de confiance de 40 %
python path/to/detect.py --weights /path/to/modelweights.pt --save_text --save_conf --conf-thres 0.40 --source 0
Si le fichier de pondérations du modèle s'intitulait "my_custom_weights.pt", alors le code devient :
python ./Users/mtraor3/yolov5/detect.py --weights ./Users/mtraor3/yolov5/my_custom_weights.pt --save_text --save_conf --conf-thres 0.40 --source 0
Arguments facultatifs
Il y a plus d'arguments facultatifs pour faire des choses comme changer le seuil de confiance, enregistrer des boîtes de prédiction recadrées, etc.
Plus d'exemples :
Enregistrement des prédictions et des scores de confiance dans un fichier .txt, avec inférence à un seuil de confiance de 40 %.