thangkt commited on
Commit
c690625
·
verified ·
1 Parent(s): 24f921a

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +39 -0
README.md ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: agpl-3.0
3
+ library_name: ultralytics
4
+ pipeline_tag: object-detection
5
+ tags: [yolo, yolov8, depgraph, torch-pruning, structured-pruning, deeppcb]
6
+ datasets: [thangkt/PCB-Prune-YOLO-DeepPCB]
7
+ metrics: [map]
8
+ ---
9
+
10
+ # PCB-Prune-YOLO P30 Direct
11
+
12
+ Strongest compression candidate in the direct DepGraph study. This
13
+ validation-selected YOLOv8n checkpoint uses structured pruning followed by the
14
+ same 50-epoch fine-tuning configuration as P10/P20. No knowledge distillation
15
+ or test-set model selection was used.
16
+
17
+ | Precision | Recall | mAP50 | mAP50-95 |
18
+ |---:|---:|---:|---:|
19
+ | 0.95324 | 0.94374 | 0.97788 | 0.75030 |
20
+
21
+ The model has 1,452,562 parameters and 1.9619 GMACs, reductions of 51.77% and
22
+ 51.83% from baseline. Its static batch-1 TensorRT FP16 engine reaches 1.754 ms
23
+ (569.97 FPS) on Tesla T4, about 1.05x the TensorRT baseline throughput, with
24
+ validation mAP50-95 0.75610.
25
+
26
+ Training used direct local group-magnitude pruning at ratio 0.30 without channel
27
+ rounding, followed by AdamW (`lr0=0.001`, `lrf=0.01`, weight decay 0.0005),
28
+ batch 64, patience 10, seed 42, AMP, and deterministic mode.
29
+
30
+ Structured pruning changes the architecture. Clone and install the
31
+ [project](https://github.com/pnthang04/PCB-Prune-YOLO) before loading:
32
+
33
+ ```python
34
+ from ultralytics import YOLO
35
+ model = YOLO("best.pt")
36
+ results = model("pcb.jpg", imgsz=640)
37
+ ```
38
+
39
+ New-process CUDA inference was verified with output `[1,10,8400]`.