102 lines
2.1 KiB
YAML
102 lines
2.1 KiB
YAML
|
# 任务定义 + 训练超参。数据包路径见 data_packs.yaml；训练用哪些包见 ML/workflow.registry.yaml active_packs
|
|||
|
|
|
|||
|
|
version: 2
|
|||
|
|
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:
|
|||
|
|
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:
|
|||
|
|
type: detect
|
|||
|
|
task_dir: addw
|
|||
|
|
nc: 4
|
|||
|
|
names: [face, eye_open, nod_face, nod_eye]
|
|||
|
|
inbox: inbox/addw
|
|||
|
|
extra_train: []
|
|||
|
|
|
|||
|
|
addw_face:
|
|||
|
|
type: pose
|
|||
|
|
task_dir: addw_face
|
|||
|
|
kpt_shape: [37, 3]
|
|||
|
|
names: {0: face}
|
|||
|
|
inbox: inbox/addw_face
|
|||
|
|
extra_train: []
|
|||
|
|
|
|||
|
|
isa:
|
|||
|
|
type: detect
|
|||
|
|
task_dir: isa
|
|||
|
|
nc: 4
|
|||
|
|
names: [indicative, prohibitory, warning, vehicle]
|
|||
|
|
inbox: inbox/isa
|
|||
|
|
extra_train: []
|
|||
|
|
|
|||
|
|
dam:
|
|||
|
|
type: detect
|
|||
|
|
task_dir: dam
|
|||
|
|
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
|
|||
|
|
extra_train: []
|
|||
|
|
|
|||
|
|
dam_0417:
|
|||
|
|
type: detect
|
|||
|
|
task_dir: dam_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_0417
|
|||
|
|
extra_train: []
|
|||
|
|
|
|||
|
|
isa_class:
|
|||
|
|
type: classify
|
|||
|
|
task_dir: isa_class
|
|||
|
|
inbox: inbox/isa_class
|
|||
|
|
extra_train: []
|