File size: 760 Bytes
fd10dcd 1b75e5e fd10dcd abfdd6f 1b75e5e fd10dcd 04dda20 3858777 c4f5fa9 1b75e5e 3858777 c4f5fa9 1b75e5e fd10dcd 41f56ca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
import gradio as gr
import cv2 as cv
from ultralytics import YOLO
# Load the YOLO model
model = YOLO('yolov8n-seg.pt')
def resize_frame(frame, width=1280):
height, width = frame.shape[:2]
new_height = int(height * width / float(width))
return cv.resize(frame, (width, new_height))
def process_frame(frame):
frame = resize_frame(frame)
results = model(frame)
segments = results[0].plot()
return segments
frame = gr.components.Video(label="Webcam Feed")
iface = gr.Interface(
fn=process_frame,
inputs=[frame],
outputs=[gr.components.Image()],
live=True,
title="YOLO Image Segmentation",
description="This application uses the YOLO model to perform image segmentation on a webcam feed.",
)
iface.launch()
|