teslatony commited on
Commit
51b4d27
·
verified ·
1 Parent(s): ecce64d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -2,8 +2,10 @@ import cv2
2
  from ultralytics import YOLO
3
  import gradio as gr
4
 
 
5
  model = YOLO("yolov8n.pt")
6
 
 
7
  def yolo_process(frame):
8
  results = model.predict(frame, imgsz=640, conf=0.25, verbose=False)
9
  annotated = results[0].plot()
@@ -11,12 +13,17 @@ def yolo_process(frame):
11
 
12
  with gr.Blocks() as demo:
13
  gr.Markdown("### 🎥 YOLOv8: распознавание объектов с веб-камеры")
14
- webcam = gr.Video(label="Веб-камера", type="numpy", streaming=True)
15
- output_video = gr.Image(label="Результат")
16
- webcam.stream(fn=yolo_process, inputs=webcam, outputs=output_video)
 
 
 
 
17
 
18
  demo.launch()
19
 
20
 
21
 
22
 
 
 
2
  from ultralytics import YOLO
3
  import gradio as gr
4
 
5
+ # Загружаем модель
6
  model = YOLO("yolov8n.pt")
7
 
8
+ # Функция обработки кадра
9
  def yolo_process(frame):
10
  results = model.predict(frame, imgsz=640, conf=0.25, verbose=False)
11
  annotated = results[0].plot()
 
13
 
14
  with gr.Blocks() as demo:
15
  gr.Markdown("### 🎥 YOLOv8: распознавание объектов с веб-камеры")
16
+
17
+ # Веб-камера
18
+ webcam = gr.Video(source="webcam", label="Веб-камера")
19
+ output_image = gr.Image(label="Результат")
20
+
21
+ # Связываем обработку с видеопотоком
22
+ webcam.change(fn=yolo_process, inputs=webcam, outputs=output_image)
23
 
24
  demo.launch()
25
 
26
 
27
 
28
 
29
+