Ce projet utilise le modèle YOLO (You Only Look Once) pour détecter les personnes, les casques et les gilets dans un flux vidéo en direct. Il utilise Flask pour créer une interface web permettant de visualiser le flux vidéo en direct avec les boîtes de détection superposées et un message de conformité en temps réel.
-
Clonez ce dépôt sur votre machine locale :
git clone https://github.com/Artemis-IA/YOLOv8_Detection/.git
-
Accédez au répertoire du projet :
cd yolov8_detection
-
Installez les dépendances nécessaires :
pip install -r requirements.txt
-
Lancez l'application Flask :
flask run
-
Accédez à l'application dans votre navigateur à l'adresse
http://localhost:5000
. -
Soumettez une image ou accédez au flux vidéo en direct pour voir la détection en action.
- Vous pouvez personnaliser les classes détectées et leurs alias dans le fichier
aliases.py
. - La logique de détection et de traitement d'image se trouve dans la fonction
detect_objects
du fichierapp.py
.
Ce projet est sous licence MIT License.