Instructions to use thangkt/PCB-Prune-YOLO-P10-Direct with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- ultralytics
How to use thangkt/PCB-Prune-YOLO-P10-Direct with ultralytics:
from ultralytics import YOLOvv8 model = YOLOvv8.from_pretrained("thangkt/PCB-Prune-YOLO-P10-Direct") source = 'http://images.cocodataset.org/val2017/000000039769.jpg' model.predict(source=source, save=True) - Notebooks
- Google Colab
- Kaggle
Upload folder using huggingface_hub
Browse files
validation_and_benchmark/benchmark/benchmark.csv
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
parameters,counted_parameters,macs,gmacs,flops_estimate,gflops_estimate,mean_latency_ms,median_latency_ms,p95_latency_ms,fps,model,model_size_mb,batch_size,imgsz,device,gpu_name,gpu_total_memory_mb,peak_gpu_memory_mb,python_version,torch_version,cuda_version,ultralytics_version
|
| 2 |
+
2416871,2416871,3269461600,3.2694616,6538923200,6.5389232,10.432576260045607,10.057338500701007,14.695106001454405,95.85360078600839,outputs/finetune_direct_fair/p10_adamw_exact/weights/best.pt,4.854181289672852,1,640,cuda:1,Tesla T4,14911.6875,41.6650390625,3.12.12,2.10.0+cu128,12.8,8.4.115
|
validation_and_benchmark/benchmark/benchmark.json
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"parameters": 2416871,
|
| 3 |
+
"counted_parameters": 2416871,
|
| 4 |
+
"macs": 3269461600,
|
| 5 |
+
"gmacs": 3.2694616,
|
| 6 |
+
"flops_estimate": 6538923200,
|
| 7 |
+
"gflops_estimate": 6.5389232,
|
| 8 |
+
"mean_latency_ms": 10.432576260045607,
|
| 9 |
+
"median_latency_ms": 10.057338500701007,
|
| 10 |
+
"p95_latency_ms": 14.695106001454405,
|
| 11 |
+
"fps": 95.85360078600839,
|
| 12 |
+
"model": "outputs/finetune_direct_fair/p10_adamw_exact/weights/best.pt",
|
| 13 |
+
"model_size_mb": 4.854181289672852,
|
| 14 |
+
"batch_size": 1,
|
| 15 |
+
"imgsz": 640,
|
| 16 |
+
"device": "cuda:1",
|
| 17 |
+
"gpu_name": "Tesla T4",
|
| 18 |
+
"gpu_total_memory_mb": 14911.6875,
|
| 19 |
+
"peak_gpu_memory_mb": 41.6650390625,
|
| 20 |
+
"python_version": "3.12.12",
|
| 21 |
+
"torch_version": "2.10.0+cu128",
|
| 22 |
+
"cuda_version": "12.8",
|
| 23 |
+
"ultralytics_version": "8.4.115"
|
| 24 |
+
}
|
validation_and_benchmark/validation/metrics_val.csv
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
scope,precision,recall,mAP50,mAP50-95,class_id,class_name
|
| 2 |
+
overall,0.9647944266359008,0.9570626861387571,0.9827348191434385,0.777360110948349,,
|
| 3 |
+
class,0.9603890425534928,0.9683794466403162,0.9808642536859331,0.7160461615598585,0.0,open
|
| 4 |
+
class,0.9376404308370303,0.9494949494949495,0.9591074875914751,0.686683095296873,1.0,short
|
| 5 |
+
class,0.9699902823757591,0.9471830985915493,0.9838488252198286,0.7776552683455351,2.0,mousebite
|
| 6 |
+
class,0.9670206723630826,0.9523809523809523,0.9889641254194397,0.7405894514575715,3.0,spur
|
| 7 |
+
class,0.9916118782098838,0.9620853080568721,0.9926456648003167,0.8818182250652284,4.0,copper
|
| 8 |
+
class,0.9621142534761563,0.9628523616679037,0.9909785581436378,0.8613684639650272,5.0,pin-hole
|
validation_and_benchmark/validation/metrics_val.json
ADDED
|
@@ -0,0 +1,59 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"split": "val",
|
| 3 |
+
"overall": {
|
| 4 |
+
"precision": 0.9647944266359008,
|
| 5 |
+
"recall": 0.9570626861387571,
|
| 6 |
+
"mAP50": 0.9827348191434385,
|
| 7 |
+
"mAP50-95": 0.777360110948349
|
| 8 |
+
},
|
| 9 |
+
"per_class": [
|
| 10 |
+
{
|
| 11 |
+
"class_id": 0,
|
| 12 |
+
"class_name": "open",
|
| 13 |
+
"precision": 0.9603890425534928,
|
| 14 |
+
"recall": 0.9683794466403162,
|
| 15 |
+
"mAP50": 0.9808642536859331,
|
| 16 |
+
"mAP50-95": 0.7160461615598585
|
| 17 |
+
},
|
| 18 |
+
{
|
| 19 |
+
"class_id": 1,
|
| 20 |
+
"class_name": "short",
|
| 21 |
+
"precision": 0.9376404308370303,
|
| 22 |
+
"recall": 0.9494949494949495,
|
| 23 |
+
"mAP50": 0.9591074875914751,
|
| 24 |
+
"mAP50-95": 0.686683095296873
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"class_id": 2,
|
| 28 |
+
"class_name": "mousebite",
|
| 29 |
+
"precision": 0.9699902823757591,
|
| 30 |
+
"recall": 0.9471830985915493,
|
| 31 |
+
"mAP50": 0.9838488252198286,
|
| 32 |
+
"mAP50-95": 0.7776552683455351
|
| 33 |
+
},
|
| 34 |
+
{
|
| 35 |
+
"class_id": 3,
|
| 36 |
+
"class_name": "spur",
|
| 37 |
+
"precision": 0.9670206723630826,
|
| 38 |
+
"recall": 0.9523809523809523,
|
| 39 |
+
"mAP50": 0.9889641254194397,
|
| 40 |
+
"mAP50-95": 0.7405894514575715
|
| 41 |
+
},
|
| 42 |
+
{
|
| 43 |
+
"class_id": 4,
|
| 44 |
+
"class_name": "copper",
|
| 45 |
+
"precision": 0.9916118782098838,
|
| 46 |
+
"recall": 0.9620853080568721,
|
| 47 |
+
"mAP50": 0.9926456648003167,
|
| 48 |
+
"mAP50-95": 0.8818182250652284
|
| 49 |
+
},
|
| 50 |
+
{
|
| 51 |
+
"class_id": 5,
|
| 52 |
+
"class_name": "pin-hole",
|
| 53 |
+
"precision": 0.9621142534761563,
|
| 54 |
+
"recall": 0.9628523616679037,
|
| 55 |
+
"mAP50": 0.9909785581436378,
|
| 56 |
+
"mAP50-95": 0.8613684639650272
|
| 57 |
+
}
|
| 58 |
+
]
|
| 59 |
+
}
|