teslatony commited on
Commit
5a6db17
·
verified ·
1 Parent(s): 49e7a89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -1,19 +1,20 @@
1
  import cv2
2
  from ultralytics import YOLO
3
- import fastrtc
4
  import gradio as gr
5
 
6
  model = YOLO("yolov8n.pt")
7
 
8
- class YOLOTransformer(fastrtc.VideoTransformer):
9
- def transform(self, frame):
10
- results = model.predict(source=frame, imgsz=640, conf=0.25, verbose=False)
11
- annotated = results[0].plot()
12
- return cv2.cvtColor(annotated, cv2.COLOR_RGB2BGR)
13
 
14
  with gr.Blocks() as demo:
15
  gr.Markdown("### 🎥 YOLOv8: Распознавание объектов с веб-камеры")
16
- fastrtc.webrtc_streamer(key="yolo", video_transformer_factory=YOLOTransformer)
 
 
17
 
18
  demo.launch()
19
 
 
 
1
  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()
10
+ return cv2.cvtColor(annotated, cv2.COLOR_RGB2BGR)
 
11
 
12
  with gr.Blocks() as demo:
13
  gr.Markdown("### 🎥 YOLOv8: Распознавание объектов с веб-камеры")
14
+ camera_input = gr.Camera(label="Веб-камера")
15
+ output_video = gr.Image(label="Результат")
16
+ camera_input.change(fn=yolo_process, inputs=camera_input, outputs=output_video)
17
 
18
  demo.launch()
19
 
20
+