2026-06-03 11:40:21 +08:00
|
|
|
|
# 任务定义 + 训练超参。数据包路径见 data_packs.yaml;训练用哪些包见 workflow.registry.yaml active_packs
|
|
|
|
|
|
# domain: dms=舱内驾驶员监控 | forward=前向交通标志(检测+细分类合一任务)
|
2026-05-25 16:59:59 +08:00
|
|
|
|
|
2026-06-03 11:40:21 +08:00
|
|
|
|
version: 3
|
2026-05-25 16:59:59 +08:00
|
|
|
|
dataset_root: .
|
|
|
|
|
|
|
|
|
|
|
|
split:
|
|
|
|
|
|
val_ratio: 0.1
|
|
|
|
|
|
seed: 42
|
|
|
|
|
|
resplit_after_ingest: true
|
|
|
|
|
|
min_val_per_class: 1
|
|
|
|
|
|
min_train_per_class: 1
|
|
|
|
|
|
rare_class_train_floor: 5
|
|
|
|
|
|
|
|
|
|
|
|
ingest:
|
|
|
|
|
|
sources_subdir: sources
|
|
|
|
|
|
ingested_subdir: sources/_ingested
|
|
|
|
|
|
|
|
|
|
|
|
train:
|
|
|
|
|
|
mode: full
|
|
|
|
|
|
|
|
|
|
|
|
detect:
|
|
|
|
|
|
model: yolo26n.pt
|
|
|
|
|
|
epochs: 100
|
|
|
|
|
|
lr0: 0.01
|
|
|
|
|
|
imgsz: 640
|
|
|
|
|
|
warm_start: null
|
|
|
|
|
|
epochs_continue: 50
|
|
|
|
|
|
lr0_continue: 0.001
|
|
|
|
|
|
|
|
|
|
|
|
pose:
|
|
|
|
|
|
model: yolo26n-pose.pt
|
|
|
|
|
|
epochs: 100
|
|
|
|
|
|
lr0: 0.01
|
|
|
|
|
|
imgsz: 640
|
|
|
|
|
|
warm_start: null
|
|
|
|
|
|
epochs_continue: 50
|
|
|
|
|
|
lr0_continue: 0.001
|
|
|
|
|
|
|
|
|
|
|
|
classify:
|
|
|
|
|
|
model: yolo26n-cls.pt
|
|
|
|
|
|
epochs: 50
|
|
|
|
|
|
lr0: 0.01
|
|
|
|
|
|
imgsz: 224
|
|
|
|
|
|
warm_start: null
|
|
|
|
|
|
epochs_continue: 30
|
|
|
|
|
|
lr0_continue: 0.001
|
|
|
|
|
|
|
|
|
|
|
|
tasks:
|
|
|
|
|
|
ddaw:
|
2026-06-03 11:40:21 +08:00
|
|
|
|
domain: dms
|
|
|
|
|
|
label: DDAW 疲劳检测
|
2026-05-25 16:59:59 +08:00
|
|
|
|
type: detect
|
|
|
|
|
|
task_dir: ddaw
|
|
|
|
|
|
nc: 9
|
|
|
|
|
|
names: [face, eye_open, eye_partially_open, eye_close, mouth_open, mouth_partially_open, mouth_close, side_face, nod_face]
|
|
|
|
|
|
inbox: inbox/ddaw
|
|
|
|
|
|
extra_train: []
|
|
|
|
|
|
|
|
|
|
|
|
addw:
|
2026-06-03 11:40:21 +08:00
|
|
|
|
domain: dms
|
|
|
|
|
|
label: ADDW 分心检测
|
2026-05-25 16:59:59 +08:00
|
|
|
|
type: detect
|
|
|
|
|
|
task_dir: addw
|
|
|
|
|
|
nc: 4
|
|
|
|
|
|
names: [face, eye_open, nod_face, nod_eye]
|
|
|
|
|
|
inbox: inbox/addw
|
|
|
|
|
|
extra_train: []
|
|
|
|
|
|
|
|
|
|
|
|
addw_face:
|
2026-06-03 11:40:21 +08:00
|
|
|
|
domain: dms
|
|
|
|
|
|
label: ADDW 人脸关键点
|
2026-05-25 16:59:59 +08:00
|
|
|
|
type: pose
|
|
|
|
|
|
task_dir: addw_face
|
|
|
|
|
|
kpt_shape: [37, 3]
|
|
|
|
|
|
names: {0: face}
|
|
|
|
|
|
inbox: inbox/addw_face
|
|
|
|
|
|
extra_train: []
|
|
|
|
|
|
|
|
|
|
|
|
dam:
|
2026-06-03 11:40:21 +08:00
|
|
|
|
domain: dms
|
|
|
|
|
|
label: DAM 驾驶员监控
|
|
|
|
|
|
type: multi
|
2026-05-25 16:59:59 +08:00
|
|
|
|
task_dir: dam
|
2026-06-03 11:40:21 +08:00
|
|
|
|
modes:
|
|
|
|
|
|
batch_0516:
|
|
|
|
|
|
label: 0516 批次
|
|
|
|
|
|
type: detect
|
|
|
|
|
|
subdir: batch_0516
|
|
|
|
|
|
nc: 15
|
|
|
|
|
|
names: [face, eye_open, eye_partially_open, eye_close, mouth_open, mouth_partially_open, mouth_close, side_face, nod_face, glasses, sunglasses, smoke, phone, driver, rise_face]
|
|
|
|
|
|
inbox: inbox/dam/batch_0516
|
|
|
|
|
|
extra_train: []
|
|
|
|
|
|
batch_0417:
|
|
|
|
|
|
label: 0417 批次
|
|
|
|
|
|
type: detect
|
|
|
|
|
|
subdir: batch_0417
|
|
|
|
|
|
nc: 15
|
|
|
|
|
|
names: [face, eye_open, eye_partially_open, eye_close, mouth_open, mouth_partially_open, mouth_close, side_face, nod_face, glasses, sunglasses, smoke, phone, driver, rise_face]
|
|
|
|
|
|
inbox: inbox/dam/batch_0417
|
|
|
|
|
|
extra_train: []
|
2026-05-25 16:59:59 +08:00
|
|
|
|
|
2026-06-03 11:40:21 +08:00
|
|
|
|
forward:
|
|
|
|
|
|
domain: forward
|
|
|
|
|
|
label: 前向·交通标志
|
|
|
|
|
|
type: multi
|
|
|
|
|
|
task_dir: forward
|
|
|
|
|
|
modes:
|
|
|
|
|
|
detect:
|
|
|
|
|
|
label: 粗检测(4 类)
|
|
|
|
|
|
type: detect
|
|
|
|
|
|
subdir: detect
|
|
|
|
|
|
nc: 4
|
|
|
|
|
|
names: [indicative, prohibitory, warning, vehicle]
|
|
|
|
|
|
inbox: inbox/forward/detect
|
|
|
|
|
|
extra_train: []
|
|
|
|
|
|
classify:
|
|
|
|
|
|
label: 细分类
|
|
|
|
|
|
type: classify
|
|
|
|
|
|
subdir: classify
|
|
|
|
|
|
inbox: inbox/forward/classify
|
|
|
|
|
|
extra_train: []
|
2026-05-25 16:59:59 +08:00
|
|
|
|
|
2026-06-03 11:40:21 +08:00
|
|
|
|
adas:
|
|
|
|
|
|
domain: forward
|
|
|
|
|
|
label: 前向·ADAS 7 类检测
|
|
|
|
|
|
type: detect
|
|
|
|
|
|
task_dir: adas
|
|
|
|
|
|
nc: 7
|
|
|
|
|
|
names: [Pedestrain, Car, Truck, Bus, Motor-vehicles, Tricycle, cones]
|
|
|
|
|
|
mode: detect
|
|
|
|
|
|
inbox: inbox/adas
|