| --- |
| license: mit |
| tags: |
| - pytorch |
| - onnx |
| - object-detection |
| - yolo |
| - computer-vision |
| - pcb-inspection |
| - industrial |
| --- |
| |
| # Inspection Engine |
|
|
| ## Model Description |
| Inspection Engine is a YOLO11s model fine-tuned for automated PCB (Printed Circuit Board) defect detection. It identifies manufacturing defects such as missing components, solder bridges, lifted pads, and other anomalies in real-time, enabling automated quality control in electronics manufacturing. |
|
|
| ## Model Architecture |
| - **Base Model**: YOLO11s (Small variant — optimized for speed) |
| - **Framework**: PyTorch + ONNX (for cross-platform deployment) |
| - **Task**: Object Detection (Defect Localization) |
| - **Input**: High-resolution PCB images |
|
|
| ## Training Details |
| - **Dataset**: PCB defect inspection dataset with annotated defect regions |
| - **Checkpoint**: `inspection_engine_final3` (best performing run) |
| - **Augmentations**: Mosaic, color jitter, random affine transforms |
| - **Export**: Exported to ONNX for production deployment |
|
|
| ## Files |
| | File | Description | |
| |------|-------------| |
| | `best.pt` | Best PyTorch model weights | |
| | `best.onnx` | ONNX export for cross-platform/production deployment | |
|
|
| ## Usage |
|
|
| ### PyTorch (Ultralytics) |
| ```python |
| from ultralytics import YOLO |
| from huggingface_hub import hf_hub_download |
| |
| model_path = hf_hub_download(repo_id='devanshty/Inspection-Engine', filename='best.pt') |
| model = YOLO(model_path) |
| results = model('pcb_image.jpg') |
| results[0].show() |
| ``` |
|
|
| ### ONNX Runtime |
| ```python |
| import onnxruntime as ort |
| import numpy as np |
| from huggingface_hub import hf_hub_download |
| |
| onnx_path = hf_hub_download(repo_id='devanshty/Inspection-Engine', filename='best.onnx') |
| session = ort.InferenceSession(onnx_path) |
| # Prepare input (1, 3, H, W) float32 normalized |
| input_name = session.get_inputs()[0].name |
| outputs = session.run(None, {input_name: np.zeros((1, 3, 640, 640), dtype=np.float32)}) |
| ``` |
|
|
| ## Download & Use |
|
|
| ```python |
| from huggingface_hub import hf_hub_download |
| model_path = hf_hub_download(repo_id='devanshty/Inspection-Engine', filename='best.pt') |
| ``` |
|
|
