Untitled

 avatar
user_9363972
python
13 days ago
1.3 kB
2
Indexable
Never
import cv2
from tracker import EuclidienDistTracker
tracker = EuclidienDistTracker()

cap = cv2.VideoCapture("highway.mp4")
object_detector = cv2.createBackgroundSubtractorMOG2(
    history=50, varThreshold=40)
while True:
    ret, frame = cap.read()
    roi = frame[340:720, 500:800]
    mask = object_detector.apply(roi)
    _, mask = cv2.threshold(mask, 254, 255, cv2.THRESH_BINARY)
    contours, _ = cv2.findContours(
        mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
    detections = []
    for cnt in contours:
        area = cv2.contourArea(cnt)
        if area > 100:
            x, y, w, h = cv2.boundingRect(cnt)
            detections.append([x, y, w, h])
            # cv2.rectangle(roi, (x, y), (x+w, y+h), (0, 255, 0), 2)
            # cv2.drawContours(roi, [cnt], -1, (0, 255, 0), 2)

    # print(detections)
    boxes_ids = tracker.update(detections)
    for boxes_id in boxes_ids:
        x, y, w, h, id = boxes_id
        cv2.putText(roi, str(id), (x, y - 15),
                    cv2.FONT_HERSHEY_COMPLEX, 1, (255, 0, 0), 2)
        cv2.rectangle(roi, (x, y), (x+w, y+h), (0, 255, 0), 2)
    cv2.imshow("Frame", frame)
    # cv2.imshow("mask", mask)

    key = cv2.waitKey(30)
    if key == ord("q"):
        break


cap.release()
cv2.destroyAllWindows()
Leave a Comment