import time

import cv2
from ultralytics import YOLO

if __name__ == "__main__":
    video_path1 = "kingGlory.mp4"
    cap1 = cv2.VideoCapture(video_path1)
    model = YOLO("runs/detect/train/weights/best.pt")

    while True:
        t1 = time.time()
        ret1, frame_start = cap1.read()
        if not ret1:
            break
        results = model.predict(source=frame_start)[0]
        plot_img = results.plot()
        cv2.imshow('image', plot_img)
        cv2.waitKey(1)
    cv2.destroyAllWindows()
    cap1.release()