README.md

metadata

license: other
tags:
  - object-detection
  - pcb-defect-detection
  - yolov8
  - k210
  - edge-ai
  - fine-tuned

Project_final — PCB Assembly Defect Detection (v2)

Trained for PS82 — Low-cost Optical Inspection for PCB Assembly.

Base model: YOLOv8n, fine-tuned in two phases:

1. Phase 1 (30 ep): Backbone frozen, head-only training with lr=0.01
2. Phase 2 (170 ep): Full fine-tuning with lr=0.001, cosine decay, aggressive augmentation

Input resolution: 320×320 (K210-safe)

Target deployment: Sipeed Maix Bit (Kendryte K210), via ONNX → NNCase INT8 .kmodel.

Classes

['solder_bridge', 'missing_component', 'component_misalignment', 'cold_solder_joint']

Training Improvements (v2)

• Transfer learning from saved checkpoint
• Two-phase training (freeze backbone → full fine-tune)
• Offline augmentation for underrepresented classes
• Cosine LR schedule with warmup
• Label smoothing (0.05)
• Aggressive data augmentation (mosaic, mixup, copy-paste, perspective)
• AdamW optimizer

See dataset/ATTRIBUTION.md for source dataset licensing.