File size: 1,436 Bytes
2b0f956 |
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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
---
license: mit
tags:
- yolo
- object-detection
- knowledge-distillation
- quantization
- military
- tank-detection
datasets:
- roboflow/military-object-detection
metrics:
- mAP
---
# YOLO Tank Detection Models
Military object detection models trained with Knowledge Distillation and Post-Training Quantization.
## Models
| Model | mAP50 | mAP50-95 | Size |
|-------|-------|----------|------|
| Teacher (YOLOv8m) | 86.40% | 62.78% | 49.6 MB |
| Student Fine-tuned (YOLOv8n) | 84.61% | 60.64% | 5.96 MB |
| Student KD (YOLOv8n) | 79.28% | 53.38% | ~6 MB |
| Student Quantized (INT8) | 84.31% | 60.50% | 3.20 MB |
## Usage
```python
from huggingface_hub import hf_hub_download
# Download quantized model for web deployment
model_path = hf_hub_download(
repo_id="Hunjun/yolo-tank-detection",
filename="optimized/student_quantized.onnx"
)
# Or download teacher model
teacher_path = hf_hub_download(
repo_id="Hunjun/yolo-tank-detection",
filename="teacher/yolov8m_tank_best.pt"
)
```
## Classes
- Airplane
- Helicopter
- Person
- Tank
- Vehicle
## Training
- Teacher: YOLOv8m fine-tuned on Military Dataset (20 epochs)
- Student: YOLOv8n fine-tuned / Knowledge Distillation (20 epochs)
- Quantization: INT8 dynamic quantization via ONNX Runtime
## References
- [Ultralytics YOLOv8](https://github.com/ultralytics/ultralytics)
- [Knowledge Distillation (Hinton et al., 2015)](https://arxiv.org/abs/1503.02531)
|