bdzl2/ai_project_v1
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
martin 2026-02-26 14:45:05 +08:00
parent dd931f6231
commit 63f240ac3a
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
yolo/cv_multi_model_back_video.py
View File

@ -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.cal_func import cal_canv_location_by_osd, red_line_reproject
from touying.ImageReproject_python.img_types import Point 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 import MultiYoloTrtDetectorTrackId
from yolo.detect.multi_yolo_trt_detect_track_trt10_yolo11 import MultiYoloTrtDetectorTrackId_TRT10_YOLO11
# 配置日志 # 配置日志
logging.basicConfig( logging.basicConfig(
@ -1885,6 +1886,7 @@ def init_capture_with_sei_fix(video_url: str, attempt: int = 1):
raise RuntimeError(f"无法打开RTMP流 (第{attempt}次尝试)") 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_BUFFERSIZE, BUFFER_SIZE) # 小缓冲区，实时推帧
cap.set(cv2.CAP_PROP_FOURCC, FOURCC) # 指定H264解码器 cap.set(cv2.CAP_PROP_FOURCC, FOURCC) # 指定H264解码器
cap.set(cv2.CAP_PROP_FPS, TARGET_FPS) # 同步流帧率 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和设备
mqtt = MQTTService(mqtt_pub_ip, port=mqtt_pub_port) mqtt = MQTTService(mqtt_pub_ip, port=mqtt_pub_port)
# detector = MultiYOLODetector_TrackId(model_configs) # 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}") print(f"mqtt_sub_ipmqtt_sub_ipmqtt_sub_ip {mqtt_sub_ip} {mqtt_sub_port} {mqtt_sub_topic}")

9
yolo_api.py
View File

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