File size: 1,185 Bytes
8c039c2 f5247c7 8c039c2 f5247c7 8c039c2 |
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 29 30 31 32 33 |
{
"architectures": ["DriverBehaviorModel"],
"model_type": "video-swin-transformer",
"backbone": "swin3d_t",
"num_classes": 5,
"class_names": ["정상", "졸음운전", "물건찾기", "휴대폰 사용", "운전자 폭행"],
"input_size": [3, 30, 224, 224],
"pretrained_backbone": "Kinetics-400",
"head": {
"type": "Sequential",
"layers": ["LayerNorm(768)", "Dropout(0.3)", "Linear(768, 5)"]
},
"training": {
"epoch": 7,
"accuracy": 0.9805,
"macro_f1": 0.9757,
"batch_size": 32,
"optimizer": "AdamW",
"learning_rate": 1e-3,
"weight_decay": 0.05,
"scheduler": "OneCycleLR",
"mixed_precision": "fp16",
"augmentation": ["Mixup(0.4)", "RandomResizedCrop", "HorizontalFlip", "ColorJitter", "TemporalAugmentation"]
},
"performance": {
"정상": {"precision": 0.97, "recall": 0.97, "f1": 0.97},
"졸음운전": {"precision": 1.00, "recall": 0.99, "f1": 0.99},
"물건찾기": {"precision": 0.95, "recall": 0.97, "f1": 0.96},
"휴대폰 사용": {"precision": 0.96, "recall": 0.96, "f1": 0.96},
"운전자 폭행": {"precision": 1.00, "recall": 1.00, "f1": 1.00}
}
}
|