72 lines
2.8 KiB
Python
72 lines
2.8 KiB
Python
import torch
|
|
|
|
|
|
def get_default_config():
|
|
return {
|
|
'rtmp': {
|
|
# 'url': "rtmp://123.132.248.154:6009/live/14",
|
|
'url': "rtmp://localhost:1935/live/14",
|
|
'max_reconnect_attempts': 20, # 增加重连次数
|
|
'reconnect_delay': 1, # 减少初始延迟
|
|
'buffer_size': 1,
|
|
'timeout_ms': 5000,
|
|
'gpu_decode': True # 启用硬件解码
|
|
},
|
|
'push': {
|
|
'enable_push': True,
|
|
# 'url': 'rtmp://123.132.248.154:6009/live/11',
|
|
'url': 'rtmp://localhost:1935/live/13',
|
|
'format': 'flv',
|
|
'video_codec': 'h264_nvenc' if torch.cuda.is_available() else 'libx264', # 使用硬件编码
|
|
'pixel_format': 'bgr24',
|
|
'preset': 'p1' if torch.cuda.is_available() else 'ultrafast', # NVIDIA专用预设
|
|
'framerate': 30,
|
|
'gpu_acceleration': True, # 启用硬件加速
|
|
'tune': 'll', # 低延迟模式
|
|
'zerolatency': 1, # 零延迟
|
|
'delay': 0, # 无延迟
|
|
'rc': 'cbr_ld_hq', # 恒定码率低延迟高质量
|
|
'bufsize': '500k' # 减少缓冲区大小
|
|
},
|
|
'model': {
|
|
'path': 'yolo11x.pt', # 保留大模型
|
|
'download_url': 'https://github.com/ultralytics/assets/releases/download/v8.3.0/yolov8x.pt'
|
|
},
|
|
'predict': {
|
|
'conf_thres': 0.25, # 提高置信度阈值
|
|
'iou_thres': 0.45,
|
|
'imgsz': 1280, # 初始推理尺寸
|
|
'line_width': 1,
|
|
'font_size': 18,
|
|
'font':"E:\\tsgz\\fonts\\PingFangSC-Medium.ttf",
|
|
'device': 'cuda:0' if torch.cuda.is_available() else 'cpu',
|
|
'frame_skip': 1, # 初始跳帧值
|
|
'half': True # 启用半精度推理
|
|
},
|
|
'task': {
|
|
'taskname': '', # 文件夹名称
|
|
'taskid': '',
|
|
'tag': {},
|
|
'aiid': '',
|
|
'res_api': 'http://123.132.248.154:6033/api/DaHuaAi/AddImg',
|
|
'api': 'http://123.132.248.154:6033/'
|
|
},
|
|
'mqtt': {
|
|
'enable': True, # 是否启用MQTT
|
|
'broker': '175.27.168.120', # MQTT代理地址
|
|
'port': 6011, # MQTT端口
|
|
'topic': 'thing/product/1581F8HGX254V00A0BUY/osd', # 订阅的主题
|
|
'client_id': 'yolo_detection_client', # 客户端ID------自己生成个
|
|
'username': 'sdhc', # 用户名（可选）
|
|
'password': None, # 密码（可选）
|
|
'keepalive': 60 # 保活时间
|
|
},
|
|
'minio': {
|
|
"UseSSL": False,
|
|
"Endpoint": "175.27.168.120:6013",
|
|
"AccessKey": "minioadmin",
|
|
"SecretKey": "minioadmin",
|
|
"BucketName": "test"
|
|
}
|
|
}
|