Upload 4 files
Browse files- README.md +15 -15
- config.json +7 -7
- pytorch_model.bin +1 -1
README.md
CHANGED
|
@@ -22,14 +22,14 @@ model-index:
|
|
| 22 |
name: Video Classification
|
| 23 |
metrics:
|
| 24 |
- type: accuracy
|
| 25 |
-
value: 0.
|
| 26 |
name: Accuracy
|
| 27 |
- type: f1
|
| 28 |
-
value: 0.
|
| 29 |
name: Macro F1
|
| 30 |
---
|
| 31 |
|
| 32 |
-
# Driver Behavior Detection Model (Epoch
|
| 33 |
|
| 34 |
운전자 이상행동 감지를 위한 Video Swin Transformer 기반 모델입니다.
|
| 35 |
|
|
@@ -46,16 +46,16 @@ model-index:
|
|
| 46 |
|:-----:|-------|:--------:|
|
| 47 |
| 0 | 정상 (Normal) | 0.95 |
|
| 48 |
| 1 | 졸음운전 (Drowsy Driving) | 0.99 |
|
| 49 |
-
| 2 | 물건찾기 (Reaching/Searching) | 0.
|
| 50 |
-
| 3 | 휴대폰 사용 (Phone Usage) | 0.
|
| 51 |
| 4 | 운전자 폭행 (Driver Assault) | 1.00 |
|
| 52 |
|
| 53 |
-
## Performance (Epoch
|
| 54 |
|
| 55 |
| Metric | Value |
|
| 56 |
|--------|-------|
|
| 57 |
-
| **Accuracy** |
|
| 58 |
-
| **Macro F1** | 0.
|
| 59 |
| **Validation Samples** | 1,371,062 |
|
| 60 |
|
| 61 |
## Training Configuration
|
|
@@ -100,14 +100,14 @@ with torch.no_grad():
|
|
| 100 |
- **Stride**: 15 frames
|
| 101 |
- **Resolution**: 224x224
|
| 102 |
|
| 103 |
-
##
|
| 104 |
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
|
| 108 |
-
|
| 109 |
-
|
| 110 |
-
|
| 111 |
|
| 112 |
## License
|
| 113 |
|
|
|
|
| 22 |
name: Video Classification
|
| 23 |
metrics:
|
| 24 |
- type: accuracy
|
| 25 |
+
value: 0.9735
|
| 26 |
name: Accuracy
|
| 27 |
- type: f1
|
| 28 |
+
value: 0.9666
|
| 29 |
name: Macro F1
|
| 30 |
---
|
| 31 |
|
| 32 |
+
# Driver Behavior Detection Model (Epoch 5)
|
| 33 |
|
| 34 |
운전자 이상행동 감지를 위한 Video Swin Transformer 기반 모델입니다.
|
| 35 |
|
|
|
|
| 46 |
|:-----:|-------|:--------:|
|
| 47 |
| 0 | 정상 (Normal) | 0.95 |
|
| 48 |
| 1 | 졸음운전 (Drowsy Driving) | 0.99 |
|
| 49 |
+
| 2 | 물건찾기 (Reaching/Searching) | 0.95 |
|
| 50 |
+
| 3 | 휴대폰 사용 (Phone Usage) | 0.94 |
|
| 51 |
| 4 | 운전자 폭행 (Driver Assault) | 1.00 |
|
| 52 |
|
| 53 |
+
## Performance (Epoch 5)
|
| 54 |
|
| 55 |
| Metric | Value |
|
| 56 |
|--------|-------|
|
| 57 |
+
| **Accuracy** | 97.35% |
|
| 58 |
+
| **Macro F1** | 0.9666 |
|
| 59 |
| **Validation Samples** | 1,371,062 |
|
| 60 |
|
| 61 |
## Training Configuration
|
|
|
|
| 100 |
- **Stride**: 15 frames
|
| 101 |
- **Resolution**: 224x224
|
| 102 |
|
| 103 |
+
## Training Progress
|
| 104 |
|
| 105 |
+
| Epoch | Accuracy | Macro F1 |
|
| 106 |
+
|:-----:|:--------:|:--------:|
|
| 107 |
+
| 2 | 95.15% | 0.9392 |
|
| 108 |
+
| 3 | 96.56% | 0.9568 |
|
| 109 |
+
| 4 | 96.83% | 0.9600 |
|
| 110 |
+
| **5** | **97.35%** | **0.9666** |
|
| 111 |
|
| 112 |
## License
|
| 113 |
|
config.json
CHANGED
|
@@ -11,9 +11,9 @@
|
|
| 11 |
"layers": ["LayerNorm(768)", "Dropout(0.3)", "Linear(768, 5)"]
|
| 12 |
},
|
| 13 |
"training": {
|
| 14 |
-
"epoch":
|
| 15 |
-
"accuracy": 0.
|
| 16 |
-
"macro_f1": 0.
|
| 17 |
"batch_size": 32,
|
| 18 |
"optimizer": "AdamW",
|
| 19 |
"learning_rate": 1e-3,
|
|
@@ -23,10 +23,10 @@
|
|
| 23 |
"augmentation": ["Mixup(0.4)", "RandomResizedCrop", "HorizontalFlip", "ColorJitter", "TemporalAugmentation"]
|
| 24 |
},
|
| 25 |
"performance": {
|
| 26 |
-
"정상": {"precision": 0.95, "recall": 0.
|
| 27 |
-
"졸음운전": {"precision": 0.99, "recall": 0.
|
| 28 |
-
"물건찾기": {"precision": 0.
|
| 29 |
-
"휴대폰 사용": {"precision": 0.
|
| 30 |
"운전자 폭행": {"precision": 1.00, "recall": 1.00, "f1": 1.00}
|
| 31 |
}
|
| 32 |
}
|
|
|
|
| 11 |
"layers": ["LayerNorm(768)", "Dropout(0.3)", "Linear(768, 5)"]
|
| 12 |
},
|
| 13 |
"training": {
|
| 14 |
+
"epoch": 5,
|
| 15 |
+
"accuracy": 0.9735,
|
| 16 |
+
"macro_f1": 0.9666,
|
| 17 |
"batch_size": 32,
|
| 18 |
"optimizer": "AdamW",
|
| 19 |
"learning_rate": 1e-3,
|
|
|
|
| 23 |
"augmentation": ["Mixup(0.4)", "RandomResizedCrop", "HorizontalFlip", "ColorJitter", "TemporalAugmentation"]
|
| 24 |
},
|
| 25 |
"performance": {
|
| 26 |
+
"정상": {"precision": 0.95, "recall": 0.96, "f1": 0.95},
|
| 27 |
+
"졸음운전": {"precision": 0.99, "recall": 0.99, "f1": 0.99},
|
| 28 |
+
"물건찾기": {"precision": 0.94, "recall": 0.96, "f1": 0.95},
|
| 29 |
+
"휴대폰 사용": {"precision": 0.95, "recall": 0.93, "f1": 0.94},
|
| 30 |
"운전자 폭행": {"precision": 1.00, "recall": 1.00, "f1": 1.00}
|
| 31 |
}
|
| 32 |
}
|
pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 126244047
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e6442f71cdbc76335fe751ada9eb7e9c4c6461d7beb82e52088bafa7e15107a5
|
| 3 |
size 126244047
|