| """ | |
| YOLOv11 Backbone 配置文件 | |
| 使用 YOLOv11 作为 backbone 的 YOLOP 模型配置 | |
| """ | |
| import os | |
| import platform | |
| from .default import _C as cfg | |
| # 覆盖默认配置以使用 YOLOv11 | |
| cfg.MODEL.USE_YOLOV11 = True | |
| cfg.MODEL.YOLOV11_SCALE = 's' # 'n', 's', 'm', 'l', 'x' | |
| cfg.MODEL.YOLOV11_WEIGHTS = '' # 'weights/yolo11s.pt' | |
| cfg.MODEL.FREEZE_BACKBONE = False # 是否冻结 backbone | |
| # 训练配置 | |
| cfg.TRAIN.BATCH_SIZE_PER_GPU = 8 # YOLOv11s 可以用更大的 batch size | |
| cfg.TRAIN.END_EPOCH = 200 # YOLOv11 建议 200+ epochs | |
| # 学习率配置（冻结 backbone 时可以用更大的学习率） | |
| cfg.TRAIN.LR0 = 0.01 # 冻结 backbone 时增大学习率 | |
| cfg.TRAIN.LRF = 0.01 | |
| cfg.TRAIN.WARMUP_EPOCHS = 3.0 # 减少 warmup | |
| # 损失权重（YOLOv11 适配） | |
| cfg.LOSS.MULTI_HEAD_LAMBDA = [1.0, 1.0, 1.0, 0.5, 0.5, 0.8] | |
| cfg.LOSS.BOX_GAIN = 7.5 | |
| cfg.LOSS.CLS_GAIN = 0.5 | |
| cfg.LOSS.DA_SEG_GAIN = 0.5 | |
| cfg.LOSS.LL_SEG_GAIN = 0.5 | |
| cfg.LOSS.LL_IOU_GAIN = 1.0 | |