From 910bb94939b15bbc7fe8770d7c7463f6fddb9f3e Mon Sep 17 00:00:00 2001
From: sunyugang <sun_shine1229@163.com>
Date: Sun, 28 Sep 2025 11:06:41 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8E=A8=E6=B5=81=20=E6=8B=89=E6=B5=81?=
 =?UTF-8?q?=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 algo/model.py | 7 +++++--
 main.py       | 6 +++++-
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/algo/model.py b/algo/model.py
index 99e8a5f..596e91b 100644
--- a/algo/model.py
+++ b/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
 
 
 # 开始训练回调
diff --git a/main.py b/main.py
index dbae2ac..19121bb 100644
--- a/main.py
+++ b/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
+    )