TheKnight115's picture
Create app.py
2b848f3 verified
raw
history blame
289 Bytes
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")