yolo11s_sensitive-detector

pyronear/yolo11s_sensitive-detector

Pyronear YOLO model for early wildfire smoke detection.

Release name: Sensitive Detector Latest version: v1.1.0

Each release is a git tag on this repo (e.g. v1.1.0). Pin a version with revision="v1.1.0" in hf_hub_download / snapshot_download.

Model details

Files

Usage

PyTorch (ultralytics)

from ultralytics import YOLO

model = YOLO("best.pt")
results = model.predict("image.jpg", imgsz=1024, conf=0.2, iou=0.01)
for r in results:
    print(r.boxes)  # bounding boxes + confidences

ONNX (onnxruntime)

from huggingface_hub import hf_hub_download
import onnxruntime as ort
import numpy as np
from PIL import Image

path = hf_hub_download(repo_id="pyronear/yolo11s_sensitive-detector", filename="onnx_cpu.tar.gz")
session = ort.InferenceSession(path, providers=["CPUExecutionProvider"])

img = Image.open("image.jpg").resize((1024, 1024))
x = np.array(img).transpose(2, 0, 1)[None].astype(np.float32) / 255.0
outputs = session.run(None, {session.get_inputs()[0].name: x})

NCNN

# Unzip first
tar -xzf ncnn_cpu.tar.gz

Download with huggingface_hub

from huggingface_hub import snapshot_download

local_dir = snapshot_download(repo_id="pyronear/yolo11s_sensitive-detector")  # latest
local_dir = snapshot_download(repo_id="pyronear/yolo11s_sensitive-detector", revision="v1.1.0")  # pinned

Pyronear engine (sequential smoke detection)

from pyroengine.engine import Engine

engine = Engine(
    conf_thresh=0.20,
    nb_consecutive_frames=5,
)
# feed frames one by one — engine.predict() returns a score
score = engine.predict(pil_image, cam_id="camera_01")
if score > engine.conf_thresh:
    print("Smoke detected!")

About Pyronear

Pyronear builds open-source tools for early wildfire detection.