Spaces:
Runtime error
Runtime error
import gradio as gr | |
import os | |
import cv2 | |
from ultralytics import YOLO | |
def object_detection(video): | |
model=YOLO("best.pt") | |
cap = cv2.VideoCapture(video) | |
while True: | |
ret, frame = cap.read() | |
if not ret: | |
break | |
results = model(frame) | |
for result in results: | |
box=result.boxes | |
x1, y1, x2, y2 = map(int, box.xyxy[0]) | |
print(x1, y1, x2, y2) | |
cv2.rectangle(frame, (x1, y1), (x2, y2), (0, 255, 0), 2) | |
return cv2.imshow("img", frame) | |
iface = gr.Interface(object_detection, | |
gr.Video(), | |
"playable_video", | |
examples=[ | |
os.path.join(os.path.abspath(''), | |
"cow-video-cows-mooing-and-grazing-in-a-field.mp4")], | |
cache_examples=True) | |
iface.launch(share=True) |