predict.py

import cv2
from ultralytics import YOLO

# Cargar modelo YOLOv8 entrenado
model = YOLO("/home/izaskunmz/yolo/yolov8-object-detection/runs/detect/train_yolov8s_v4/weights/best.pt")

# Abrir vídeo
video_path = "/ruta/al/video.mp4"
cap = cv2.VideoCapture(video_path)

while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break  # Si el vídeo ha terminado, salimos del bucle

    # Realizar detección en el frame
    results = model(frame)

    # Mostrar el frame con detecciones
    annotated_frame = results[0].plot()  # Dibuja las detecciones en el frame
    cv2.imshow("YOLOv8 Detección", annotated_frame)

    # Salir con la tecla "q"
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()