VenusArk/surface-contamination-detector

Surface Contamination Detector

Free Hugging Face Spaces demo for object-specific surface detection with local Ultralytics YOLO models.

Usage

• —Select a surface type.
• —Use Upload Image to upload and detect an image.
• —Use Camera Snapshot to capture one frame. Detection starts automatically after pressing the shutter, or press Detect Snapshot.
• —Use Live Webcam for automatic detection every second.
• —Phone Screen upload/snapshot accepts close-up screen images. Live Webcam first locates a phone, then detects defects only inside it to avoid face/background false positives.
• —Upload detection starts automatically; the detect buttons can also rerun it.
• —Review the annotated image, counts, and confidence table.

Model

The app loads a separate model for each surface type:

models/lens.pt
models/phone_screen.pt
models/glass_panel.pt
models/plastic_part.pt

Missing surface models show a clear message instead of using the wrong model.

Phone Screen uses a two-stage pipeline:

1. 1.phone_detector.pt locates the phone.
2. 2.phone_screen.pt detects Oil, Scratch, and Stain only inside the phone box.