diff --git a/yolo/cv_multi_model_back_video.py b/yolo/cv_multi_model_back_video.py index 0086328..ab83240 100644 --- a/yolo/cv_multi_model_back_video.py +++ b/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}") diff --git a/yolo_api.py b/yolo_api.py index 919e228..338e4c7 100644 --- a/yolo_api.py +++ b/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: