Table 5.
Model | Pre-Training | Unfreeze | Custom Head | Training Accuracy | Test Accuracy |
---|---|---|---|---|---|
DaViT | ✓ | ✗ | ✗ | 97.0 | 91.7 |
RailTrack-DaViT | ✗ | ✗ | ✓ | 49.2 | 50.0 |
✗ | ✓ | ✓ | 50.7 | 50.0 | |
✓ | ✗ | ✓ | 99.4 | 97.5 | |
✓ | ✓ | ✗ | 99.5 | 97.5 | |
✓ | ✓ | ✓ | 99.7 | 98.8 |
Model | Pre-Training | Unfreeze | Custom Head | Training Accuracy | Test Accuracy |
---|---|---|---|---|---|
DaViT | ✓ | ✗ | ✗ | 97.0 | 91.7 |
RailTrack-DaViT | ✗ | ✗ | ✓ | 49.2 | 50.0 |
✗ | ✓ | ✓ | 50.7 | 50.0 | |
✓ | ✗ | ✓ | 99.4 | 97.5 | |
✓ | ✓ | ✗ | 99.5 | 97.5 | |
✓ | ✓ | ✓ | 99.7 | 98.8 |