Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import cv2 | |
| import numpy as np | |
| import PIL | |
| from ultralytics import YOLO | |
| from camera_input_live import camera_input_live | |
| # Load YOLO model (Download "yolov8n.pt" first if not available) | |
| model = YOLO("yolov8n.pt") | |
| st.title("🎥 YOLO Object Detection Live Stream") | |
| image = camera_input_live() | |
| if image: | |
| uploaded_image = PIL.Image.open(image) | |
| #uploaded_image_cv = cv2.cvtColor(np.array(uploaded_image), cv2.COLOR_RGB2BGR) | |
| result = model(uploaded_image) | |
| img_plot = result[0].plot() | |
| st.image(img_plot, channels = "BGR") |