2026.6.9小更新

infer_video_direct.py

@@ -10,6 +10,7 @@ from ais_bench.infer.interface import InferSession
 # ==================== 配置区域 ======================
 MODEL_PATH = "./om/yolo26n-balloon.om"          # YOLO26 模型路径
 INPUT_VIDEO = "./asset/test_video.mp4"          # 输入的测试视频
+RTSP_URL = "rtsp://192.168.144.25:8554/main.264"
 CONF_THRESHOLD = 0.25                           # 置信度阈值
 
 # -------------- 无线图传网络配置 --------------
@@ -106,14 +107,16 @@ def draw_boxes(img, detections):
 
 def main():
     detector = YOLO26UAVInfer(MODEL_PATH)
-    cap = cv2.VideoCapture(INPUT_VIDEO)
     from control import UAVController
-
     uav = UAVController()
-
+    os.environ['OPENCV_FFMPEG_CAPTURE_OPTIONS'] = 'rtsp_transport;tcp'
+    cap = cv2.VideoCapture(RTSP_URL,cv2.CAP_FFMPEG)
     if not cap.isOpened():
-        print("❌ 无法打开视频源")
-        return
+        cap = cv2.VideoCapture(INPUT_VIDEO)
+        print("❌ 网口连接失败")
+        if not cap.isOpened():
+            print("❌ 无法打开视频源")
+            return
 
     # 启动后台无线电图传线程
     sender_thread = threading.Thread(target=udp_stream_sender_thread, daemon=True)