KAHRAMAN42's picture
Update app.py
6e7b469 verified
raw
history blame
940 Bytes
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)