针对缺少图传模块，引起的进程挂起问题，优化读流时长参数

yolo/cv_multi_model_back_video.py

@@ -33,6 +33,7 @@ from middleware.read_srt import parse_srt_file
 from touying.ImageReproject_python.cal_func import cal_canv_location_by_osd, red_line_reproject
 from touying.ImageReproject_python.img_types import Point
 from yolo.detect.multi_yolo_trt_detect_track import MultiYoloTrtDetectorTrackId
+from yolo.detect.multi_yolo_trt_detect_track_trt10_yolo11 import MultiYoloTrtDetectorTrackId_TRT10_YOLO11
 
 # 配置日志
 logging.basicConfig(
@@ -1885,6 +1886,7 @@ def init_capture_with_sei_fix(video_url: str, attempt: int = 1):
         raise RuntimeError(f"无法打开RTMP流 (第{attempt}次尝试)")
 
     # 设置核心参数
+    cap.set(cv2.CAP_PROP_READ_TIMEOUT_MSEC, 25000) #设置25s拉流超时，是为了规避类似彭州水务没加图传模块，飞机太远就拉流失败
     cap.set(cv2.CAP_PROP_BUFFERSIZE, BUFFER_SIZE)  # 小缓冲区，实时推帧
     cap.set(cv2.CAP_PROP_FOURCC, FOURCC)  # 指定H264解码器
     cap.set(cv2.CAP_PROP_FPS, TARGET_FPS)  # 同步流帧率
@@ -4004,7 +4006,8 @@ async def start_rtmp_processing(video_url: str, task_id: str, model_configs: Lis
         # 初始化MQTT和设备
         mqtt = MQTTService(mqtt_pub_ip, port=mqtt_pub_port)
         # detector = MultiYOLODetector_TrackId(model_configs)
-        detector = MultiYoloTrtDetectorTrackId(model_configs)
+        # detector = MultiYoloTrtDetectorTrackId(model_configs)
+        detector = MultiYoloTrtDetectorTrackId_TRT10_YOLO11(model_configs)
 
         print(f"mqtt_sub_ipmqtt_sub_ipmqtt_sub_ip {mqtt_sub_ip} {mqtt_sub_port} {mqtt_sub_topic}")
 

yolo_api.py

@@ -67,7 +67,14 @@ DB_CONFIG = {
     "host": "8.137.54.85",
     "port": "5060"
 }
-
+# 什邡数据库
+# DB_CONFIG = {
+#     "dbname": "postgres",
+#     "user": "postgres",
+#     "password": "root",
+#     "host": "222.213.91.11",
+#     "port": "5061"
+# }
 
 # 配置类
 class Config: