Upload 6 files

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.9-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+CMD ["python", "app.py"]

__init__.py

@@ -0,0 +1 @@
+# This file is intentionally left blank.

app.py

@@ -0,0 +1,5 @@
+# filepath: /Users/bahakizil/Desktop/rstm/yolo-rtmp-detection/app.py
+from app.main import iface
+
+if __name__ == "__main__":
+    iface.queue().launch()

main.py

@@ -0,0 +1,53 @@
+import cv2
+import gradio as gr
+import PIL.Image
+import time
+from ultralytics import YOLO
+
+# Initialize model with lower confidence threshold (0.2)
+model = YOLO("models/yolo11n.pt")
+model.conf = 0.2  # Set confidence threshold to 0.2
+
+# Default RTMP URL
+default_rtmp_url = "rtmp://flvcloudhkwa.ezvizlife.com:1935/v3/openlive/E56405322_1_1?expire=1803836768&id=819385674427736064&c=1cd321224b&t=c1eea0c9fe19262cbc4cb8d9523389e0eef1e51981a6ce8ddf8275e6ec4067b9&ev=100"
+
+def stream_yolo(rtmp_url):
+    if not rtmp_url:
+        rtmp_url = default_rtmp_url
+    
+    cap = cv2.VideoCapture(rtmp_url)
+    if not cap.isOpened():
+        print("Video stream açılırken hata oluştu.")
+        yield PIL.Image.new('RGB', (640, 480), color='red')
+        return
+        
+    while True:
+        ret, frame = cap.read()
+        if not ret:
+            print("Frame alınamadı, akış sona erdi.")
+            break
+        
+        results = model(frame)
+        # Access the first result from the results list
+        # Then render it or just use plotted directly
+        annotated_frame = results[0].plot()
+        annotated_frame = cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB)
+        image = PIL.Image.fromarray(annotated_frame)
+        yield image
+        time.sleep(0.03)
+
+iface = gr.Interface(
+    fn=stream_yolo,
+    inputs=gr.Textbox(
+        label="RTMP URL", 
+        placeholder="Enter your RTMP URL here...",
+        value=default_rtmp_url
+    ),
+    outputs=gr.Image(type="pil", label="Canlı YOLO Tespiti"),
+    live=True,
+    title="RTMP Akışından Canlı YOLO Tespiti",
+    description="Kendi CCTV akışınızı izlemek için RTMP URL'sini girin. YOLOv11n, 0.2 güven eşiği ile çalışır."
+)
+
+if __name__ == "__main__":
+    iface.launch()

requirements.txt

@@ -0,0 +1,7 @@
+import gradio
+import opencv-python
+import ultralytics
+import numpy
+import pillow
+import python-dotenv
+import requests

video_utils.py

@@ -0,0 +1,17 @@
+def extract_frames_from_stream(rtmp_url):
+    cap = cv2.VideoCapture(rtmp_url)
+    if not cap.isOpened():
+        raise ValueError("Unable to open video stream.")
+    
+    while True:
+        ret, frame = cap.read()
+        if not ret:
+            break
+        yield frame
+    
+    cap.release()
+
+def process_frame(frame, model):
+    results = model(frame)
+    annotated_frame = results.render()[0]
+    return annotated_frame