Views
No views yet
| ID | Class | Type |
|---|---|---|
| 0 | normal | Compliant |
| 1 | MORE_THAN_TWO_PERSONS | Violation |
| 2 | USING_MOBILE | Violation |
| 3 | WITH_HELMET | Compliant |
| 4 | WITHOUT_HELMET | Violation |
1from ultralytics import YOLO
2model = YOLO("best.pt")
3results = model.predict("image.jpg", conf=0.5)