Spaces:
Sleeping
Sleeping
from ultralytics import YOLO | |
model = YOLO("yolov8m.pt") | |
cap = cv2.VideoCapture(tfile.name) | |
while cap.isOpened(): | |
ret, frame = cap.read() | |
if not ret: | |
break | |
results = model(frame) | |
annotated_frame = results[0].plot() | |
stframe.image(annotated_frame, channels="BGR") |