推流 拉流功能

algo/model.py

@@ -1,6 +1,9 @@
-from ultralytics import YOLO
+import cv2
+import time
+import subprocess
+import numpy as np
 from pathlib import Path
-import cv2, subprocess, numpy as np, time
+from ultralytics import YOLO
 
 
 # 开始训练回调

main.py

@@ -3,4 +3,8 @@ from algo import YoloModel
 
 if __name__ == '__main__':
     model = YoloModel('yolo11n.pt')
-    model.detect('rtsp://admin:xkrs0425@172.16.20.15:554/D6/main/av_stream', 'rtmp://localhost/live/stream', 15)
+    model.detect(
+        'rtsp://admin:xkrs0425@172.16.20.15:554/D6/main/av_stream',
+        'rtmp://localhost/live/stream',
+        15
+    )