YOLOv11 Detection

Detection of the following objects using yolo11s.pt as base model

Car
Motorcycle
Helmet
Person

The annotation was done using Label Studio

See Report

Training

sh init.sh                      # To setup python venv
sh get_data.sh                  # Get the images
.env/bin/python3 train.py

To annotate dataset, use Label Studio: .env/bin/pip install label-studio (or even with system wide pip install)

Testing

Drop media into test folder

mkdir test                      # Drop files here
.env/bin/python3 test.py

Results are in runs folder

App.py

Using app.py, you can run on camera or any media file. It also display number of each object detected. Run .env/bin/python3 app.py -h for help