CoolFace
Modelpublic

Beehzod/yolov26-fire-smoke-detection

sourceHugging Facecc-by-4.0updated 1mo agoView on Hugging Face
0likes206downloads
Model Card

yolov26-fire-smoke-detection

Fine-tuned from `yolov26-fire-detection` (YOLOv26-S) using Ultralytics.

Classes

  • —Fire
  • —Smoke

Dataset

  • —Images: 7673 train / 925 valid / 960 test
  • —Check the dataset page above for its license -- not necessarily the same as this repo's license field, which reflects the base model's license.

Training

ParameterValue
Epochs100
Image size640
Batch size16
OptimizerAdamW
Initial LR (lr0)0.001
Patience (early stop)20

Results (held-out validation split)

MetricValue
mAP500.9177
mAP50-950.8299
Precision0.9167
Recall0.8958

Usage

python
from ultralytics import YOLO

model = YOLO("best.pt")
results = model.predict("image.jpg", conf=0.25)
for r in results:
    for box in r.boxes:
        print(model.names[int(box.cls[0])], float(box.conf[0]))