Upload trained YOLOv8 PCB model
Browse files- README.md +14 -0
- best.pt +3 -0
- 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'}
|