Files
HSAP/datasets/dms/datasets.registry.yaml

133 lines
3.0 KiB
YAML
Raw Normal View History

# 任务定义 + 训练超参。数据包路径见 data_packs.yaml训练用哪些包见 workflow.registry.yaml active_packs
# domain: dms=舱内驾驶员监控 | forward=前向交通标志(检测+细分类合一任务)
version: 3
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:
domain: dms
label: DDAW 疲劳检测
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:
domain: dms
label: ADDW 分心检测
type: detect
task_dir: addw
nc: 4
names: [face, eye_open, nod_face, nod_eye]
inbox: inbox/addw
extra_train: []
addw_face:
domain: dms
label: ADDW 人脸关键点
type: pose
task_dir: addw_face
kpt_shape: [37, 3]
names: {0: face}
inbox: inbox/addw_face
extra_train: []
dam:
domain: dms
label: DAM 驾驶员监控
type: multi
task_dir: dam
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: []
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: []
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