File size: 1,284 Bytes
1f10653 15ba09b 3bd709e 9f04171 3bd709e 9f04171 3bd709e 9f04171 3bd709e 9f04171 3bd709e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | ---
pipeline_tag: object-detection
tags:
- fire
- smoke
- safety
- pytorch
base_model:
- Ultralytics/YOLO26
---
# Safety Detection
A fine-tuned YOLO model for detecting fire and smoke in images and video streams, built for real-time safety monitoring.
## Model Details
- **Architecture:** YOLOv26 (fine-tuned)
- **Framework:** PyTorch
- **Epochs:** 52
- **Experiment Tracking:** ClearML
## Classes
| ID | Label |
|----|-------|
| 0 | fire |
| 1 | smoke |
## Dataset
Fine-tuned on the [Home Fire Dataset](https://www.kaggle.com/datasets/pengbo00/home-fire-dataset) from Kaggle.
- **Training Logs:** [ClearML Experiment](https://app.clear.ml/projects/bffe65b5fe1649dd9d202e181ba92fe0/tasks/f57871573c9d4d969dd5867004857d99/scalars)
## Evaluation Metrics
| Metric | Value |
|-----------|-------|
| mAP@50 | 0.930 |
| mAP@50-95 | 0.626 |
| Precision | 0.913 |
| Recall | 0.891 |
## Usage
```python
from ultralytics import YOLO
model = YOLO("path/to/model.pt")
results = model("image.jpg")
```
## Limitations
- Trained on home fire scenarios — performance may degrade in industrial or outdoor environments
- Detection confidence decreases at stricter IoU thresholds (mAP@50-95: 0.626)
- Not validated for production safety-critical systems without further testing |