Yolo 생성된 모델을 이용해서 객체 추적하기

[주인장]

import cv2

import math

from ultralytics import YOLO

model = YOLO("models/scooter/scooter_best.pt")

print(model.names)

# webcamera = cv2.VideoCapture(0)

cap = cv2.VideoCapture("static/video/scooter/scooter5.mp4")

cap.set(cv2.CAP_PROP_FRAME_WIDTH, 1920)

cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 1080)

classNames = ["yes", "no" "scooter", "person"]

while True:

    success, img = cap.read()

    results = model(img, stream=True)

    # coordinates

    for r in results:

        boxes = r.boxes

        for box in boxes:

            # bounding box

            x1, y1, x2, y2 = box.xyxy[0]

            x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)  # convert to int values

            # put box in cam

            cv2.rectangle(img, (x1, y1), (x2, y2), (255, 0, 0), 1)

            # confidence

            confidence = math.ceil((box.conf[0] * 100)) / 100

            # print("Confidence --->", confidence)

            # class name

            cls = int(box.cls[0])

            # print("Class name -->", classNames[cls])

            # object details

            org = [x1, y1]

            font = cv2.FONT_HERSHEY_SIMPLEX

            fontScale = 1

            color = (255, 0, 0)

            thickness = 2

            cv2.putText(img, classNames[cls], org, font, fontScale, color, thickness)

    cv2.imshow("Webcam", img)

    if cv2.waitKey(1) == ord("q"):

        break

cap.release()

cv2.destroyAllWindows()