shadrack20s/ai-inspector-callout-detection

callout_v2_yolo11s-obb

YOLO11s-OBB model finetuned for engineering drawing callout detection.

Classes

• Hole (idx 0) — diameter callouts
• TappedHole (idx 1) — thread callouts
• Fillet (idx 4) — radius callouts
• Chamfer (idx 5) — chamfer callouts

Training

• Base model: yolo11s-obb.pt
• Dataset: Roboflow ai-inspector-callout-detection v3
• Epochs: 150 (with early stopping)
• Image size: 1024
• Augmentation: Drawing-safe (no flip, no hue/sat)

Usage

from ultralytics import YOLO
model = YOLO("hf://shadrack20s/ai-inspector-callout-detection/callout_v2_yolo11s-obb_best.pt")
results = model("drawing.png")