1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
| # 训练配置
| data:
| annotations_jsonl: ../data/annotations.jsonl
| frames_dir: ../data/frames
| labels_csv: ../data/labels.csv
|
| sampling:
| sample_fps: 1.0
| positive_window_sec: 3.0
| other_downsample_ratio: 2
|
| model:
| backbone: mobilenet_v3_small
| image_size: 224
| pos_weight: null
|
| train:
| batch_size: 32
| epochs: 40
| lr: 0.0005
| early_stop_patience: 8
| num_workers: 0
| freeze_backbone: true
|
| export:
| opset: 17
| quantize: true
| output_dir: ../models
|
|
