CoolFace
Modelpublic

sanjeevafk/glasseye-yolo

sourceHugging Facemitupdated 1mo agoView on Hugging Face
0likes38downloads
Model Card

GlassEye YOLOv8n (BFDD + CUBIT Combined Checkpoint)

glasseye-yolo-bfdd-cubit-v1 is a fine-tuned YOLOv8n model built for autonomous building façade defect detection within the GlassEye inspection pipeline.

It is trained on a combined dataset of real-world façade defect imagery from BFDD (Building Façade Defect Dataset) and CUBIT (Concrete Ultrasound / Building Inspection dataset), unified under a single binary defect class.

Model Summary

  • —Base Architecture: YOLOv8n (yolov8n.pt, 3.2M parameters)
  • —Framework: PyTorch (torch 2.13+cu130) / Ultralytics 8.4.120
  • —Class Map: 0: defect (binary classification for façade surface and structural defects)
  • —Input Resolution: 320 × 320 px
  • —Model Checkpoint Size: 6.2 MB (best.pt)
  • —SHA-256: df78a4f90c776d8f122ec02972a193f91b1f4fc3916f1d78a4f51bb2e05445cf

Benchmark & Performance

Evaluated strictly across three separate test sets using standard IoU thresholds (IoU = 0.50, confidence = 0.20, imgsz = 320):

1. BFDD Held-Out Test Benchmark (Primary Ground-Truth)

149 untouched images, 1,033 mask-derived ground truth boxes.

Model CheckpointmAP@50mAP@50-95Recall
Synthetic Baseline (glasseye-yolo-v1)0.01310.00990.0165
BFDD-only (glasseye-yolo-real-bfdd-v1)0.09400.04290.1413
BFDD + CUBIT (`glasseye-yolo-bfdd-cubit-v1`)0.15120.07070.2014
Improvement: BFDD+CUBIT achieves +60.8% mAP@50, +64.8% mAP@50-95, and +42.5% Recall over the BFDD-only model on untouched held-out building façade images.

2. CUBIT Test Benchmark

701 images, 5,085 derived polygon boxes.

Model CheckpointFull Test mAP@50Full Test RecallFar-Subset mAP@50 (≥10 frames dist)
Synthetic Baseline0.00310.00690.0000
BFDD-only0.02110.03500.0164
BFDD + CUBIT0.19900.20610.1279

3. UAV2K Test Benchmark (Aerial Out-of-Domain)

200 images, 527 derived boxes (high-resolution aerial drone captures).

Model CheckpointmAP@50RecallTrue Positives
Synthetic Baseline0.00000.00000
BFDD-only0.00000.00000
BFDD + CUBIT0.01320.01337

Training Details

  • —Dataset: data/bfdd_cubit_binary_v1 (Train: 1,299 images [600 BFDD + 699 CUBIT val-as-train]; Val: 89 BFDD images; Test: 850 images).
  • —Optimizer: AdamW
  • —Epochs: 32
  • —Batch Size: 4
  • —Image Size: 320 px
  • —Seed: 20260815 (deterministic)
  • —Augmentations: Horizontal flip (0.5), HSV-H (0.01), HSV-S (0.2), HSV-V (0.15), Mosaic (0.0).

Quickstart & Usage

1. Installation

bash
pip install ultralytics torch huggingface_hub

2. Run Inference in Python

python
from ultralytics import YOLO
from huggingface_hub import hf_hub_download

# Download model from Hugging Face
model_path = hf_hub_download(
    repo_id="sanjeevafk/glasseye-yolo-bfdd-cubit-v1",
    filename="best.pt"
)

# Load model
model = YOLO(model_path)

# Predict on an image or video
results = model.predict(
    source="path/to/facade_image.jpg",
    conf=0.20,
    iou=0.45,
    imgsz=320,
    save=True
)

for r in results:
    for box in r.boxes:
        print(f"Detected {model.names[int(box.cls)]} at {box.xyxy.tolist()} (conf: {float(box.conf):.3f})")

3. Ultralytics CLI

bash
# Run detection directly
yolo predict model=best.pt source="https://example.com/facade.jpg" conf=0.20 imgsz=320

Project Context

This model is integrated into GlassEye, a deterministic façade-inspection pipeline featuring YOLO object detection, panel projection mapping, event logging, and advisory VLM second opinions.