santhosh1305 commited on
Commit
f87d2e8
·
verified ·
1 Parent(s): 80fa269

Upload trained YOLOv8 PCB model

Browse files
Files changed (3) hide show
  1. README.md +14 -0
  2. best.pt +3 -0
  3. pcb_defects.yaml +13 -0
README.md ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # PCB Defect Detection (YOLOv8)
2
+
3
+ Trained YOLOv8 model for PCB defects.
4
+
5
+ **Files:**
6
+ - best.pt : YOLOv8 weights
7
+ - pcb_defects.yaml : dataset config (classes)
8
+
9
+ **Usage (example):**
10
+ ```python
11
+ from ultralytics import YOLO
12
+ model = YOLO("https://huggingface.co/<USERNAME>/pcb-defect-detector/resolve/main/best.pt")
13
+ model("pcb.jpg")
14
+ ```
best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9adaae5a5cd29f5f721715b26be248ed667edea24acb9f4819ca186abd4ce2e0
3
+ size 22518954
pcb_defects.yaml ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # PCB Defects Dataset YAML configuration
2
+ # Total images: 10,668 (Train: 8,534, Val: 2,134)
3
+ # Defect classes: 6
4
+
5
+ path: /content/PCB_YOLO_Dataset # dataset root dir
6
+ train: images/train # train images (relative to 'path')
7
+ val: images/val # val images (relative to 'path')
8
+
9
+ # Number of classes
10
+ nc: 6
11
+
12
+ # Class names
13
+ names: {0: 'missing_hole', 1: 'mouse_bite', 2: 'open_circuit', 3: 'short', 4: 'spur', 5: 'spurious_copper'}