Hunjun commited on
Commit
2b0f956
·
verified ·
1 Parent(s): 9520761

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +64 -0
README.md ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - yolo
5
+ - object-detection
6
+ - knowledge-distillation
7
+ - quantization
8
+ - military
9
+ - tank-detection
10
+ datasets:
11
+ - roboflow/military-object-detection
12
+ metrics:
13
+ - mAP
14
+ ---
15
+
16
+ # YOLO Tank Detection Models
17
+
18
+ Military object detection models trained with Knowledge Distillation and Post-Training Quantization.
19
+
20
+ ## Models
21
+
22
+ | Model | mAP50 | mAP50-95 | Size |
23
+ |-------|-------|----------|------|
24
+ | Teacher (YOLOv8m) | 86.40% | 62.78% | 49.6 MB |
25
+ | Student Fine-tuned (YOLOv8n) | 84.61% | 60.64% | 5.96 MB |
26
+ | Student KD (YOLOv8n) | 79.28% | 53.38% | ~6 MB |
27
+ | Student Quantized (INT8) | 84.31% | 60.50% | 3.20 MB |
28
+
29
+ ## Usage
30
+
31
+ ```python
32
+ from huggingface_hub import hf_hub_download
33
+
34
+ # Download quantized model for web deployment
35
+ model_path = hf_hub_download(
36
+ repo_id="Hunjun/yolo-tank-detection",
37
+ filename="optimized/student_quantized.onnx"
38
+ )
39
+
40
+ # Or download teacher model
41
+ teacher_path = hf_hub_download(
42
+ repo_id="Hunjun/yolo-tank-detection",
43
+ filename="teacher/yolov8m_tank_best.pt"
44
+ )
45
+ ```
46
+
47
+ ## Classes
48
+
49
+ - Airplane
50
+ - Helicopter
51
+ - Person
52
+ - Tank
53
+ - Vehicle
54
+
55
+ ## Training
56
+
57
+ - Teacher: YOLOv8m fine-tuned on Military Dataset (20 epochs)
58
+ - Student: YOLOv8n fine-tuned / Knowledge Distillation (20 epochs)
59
+ - Quantization: INT8 dynamic quantization via ONNX Runtime
60
+
61
+ ## References
62
+
63
+ - [Ultralytics YOLOv8](https://github.com/ultralytics/ultralytics)
64
+ - [Knowledge Distillation (Hinton et al., 2015)](https://arxiv.org/abs/1503.02531)