Valve Detection YOLOv8s β€” Gas Infrastructure Inspection

Model Description

YOLOv8s model trained to detect 4 types of valves in underground gas valve well inspection images. Trained using an iterative pseudo-labeling strategy starting from just 30 manually annotated images, achieving 92.95% mAP50 through 10 rounds of self-improvement.

Performance

Metric Value
mAP50 92.95%
mAP50-95 72.8%
Model Size 21.5 MB (PyTorch)
Inference Speed ~8ms per image (MPS)
Training Rounds 10 (iterative pseudo-labeling)

Detectable Valve Types (4 Classes)

Class Description
gate_valve Gate valve (ι—Έι˜€)
globe_valve Globe valve (ζˆͺζ­’ι˜€)
ball_valve Ball valve (ηƒι˜€)
other_valve Other valve types (ε…Άδ»–)

Usage

from ultralytics import YOLO

model = YOLO("lg227210/valve-detection-yolov8s")
results = model("inspection_photo.jpg")

for result in results:
    for box in result.boxes:
        cls = int(box.cls)
        conf = float(box.conf)
        print(f"Valve: {model.names[cls]}, Confidence: {conf:.2f}")

Training Methodology

This model was trained using iterative pseudo-labeling:

  1. Start with 30 manually annotated images
  2. Train initial model (R1)
  3. Use model to generate pseudo-labels for unlabeled images
  4. Retrain with expanded dataset
  5. Repeat for 10 rounds, each time improving accuracy
Round Training Images mAP50
R1 30 34.5%
R5 ~2,000 78.2%
R10 ~30,000 92.95%

Part of the Inspection Pipeline

This model is part of a 3-stage pipeline:

  1. Valve Detection β€” This model (mAP50 = 92.95%)
  2. Anomaly Detection β€” lg227210/anomaly-detection-yolov8s (mAP50 = 48.2%)
  3. Anomaly Classification β€” EfficientNet-B0 (74% accuracy, severity scoring)

Available Formats

  • PyTorch (.pt) β€” Default
  • ONNX (.onnx) β€” Cross-platform deployment
  • CoreML (.mlpackage) β€” iOS/macOS
  • TorchScript (.torchscript) β€” Embedded systems

Dataset

Trained on 861,000+ real underground gas valve well inspection images (2592x1944 resolution, 973K total, 87.9% quality pass rate).

Links

Commercial Licensing

For commercial use, contact for licensing options. Custom model development services available starting at $500.


Built on Apple M4 Mac Mini with PyTorch MPS acceleration. Total training: ~50 hours across 10 rounds of iterative pseudo-labeling.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Space using lg227210/valve-detection-yolov8s 1