Upload 3 files

hdv.py

@@ -0,0 +1,36 @@
+
+from ultralytics import YOLO
+
+# Initialize video capture
+video_path = "Media/traffic.mp4"
+cap = cv2.VideoCapture(video_path)
+
+# Load YOLO model with custom weights
+model = YOLO("Weights/best.pt")
+
+# Define class names
+classNames = ['With Helmet', 'Without Helmet']
+
+# For the use of Webcam
+# Open the webcam (use 0 for the default camera, or 1, 2, etc. for additional cameras)
+# cap = cv2.VideoCapture(0)
+
+while True:
+    success, img = cap.read()
+    results = model(img, stream=True)
+    for r in results:
+        boxes = r.boxes
+        for box in boxes:
+            x1, y1, x2, y2 = box.xyxy[0]
+            x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
+
+            w, h = x2 - x1, y2 - y1
+            cvzone.cornerRect(img, (x1, y1, w, h))
+            conf = math.ceil((box.conf[0] * 100)) / 100
+            cls = int(box.cls[0])
+
+            cvzone.putTextRect(img, f'{classNames[cls]} {conf}', (max(0, x1), max(35, y1)), scale=1, thickness=1)
+
+    cv2.imshow("Image", img)
+    if cv2.waitKey(1) & 0xFF == ord('q'):
+        break

helmet_detector.py

@@ -0,0 +1,43 @@
+import cv2
+import math
+import cvzone
+from ultralytics import YOLO
+
+# Load YOLO model with custom weights
+yolo_model = YOLO("Weights/best.pt")
+
+# Define class names
+class_labels = ['With Helmet', 'Without Helmet']
+
+# Load the image
+image_path = "Media/riders_1.jpg"
+img = cv2.imread(image_path)
+
+# Perform object detection
+results = yolo_model(img)
+
+# Loop through the detections and draw bounding boxes
+for r in results:
+    boxes = r.boxes
+    for box in boxes:
+        x1, y1, x2, y2 = box.xyxy[0]
+        x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
+
+        w, h = x2 - x1, y2 - y1
+        cvzone.cornerRect(img, (x1, y1, w, h))
+        conf = math.ceil((box.conf[0] * 100)) / 100
+        cls = int(box.cls[0])
+
+        if conf > 0.1:
+            cvzone.putTextRect(img, f'{class_labels[cls]} {conf}', (x1, y1 - 10), scale=0.8, thickness=1, colorR=(255, 0, 0))
+
+# Display the image with detections
+cv2.imshow("Image", img)
+
+# Close window when 'q' button is pressed
+while True:
+    if cv2.waitKey(1) & 0xFF == ord('q'):
+        break
+
+cv2.destroyAllWindows()
+cv2.waitKey(1)

requiremets.txt

@@ -0,0 +1,2 @@
+yolov9pip==0.0.4
+huggingface_hub