Update README.md
Browse files
README.md
CHANGED
|
@@ -1,6 +1,4 @@
|
|
| 1 |
---
|
| 2 |
-
language:
|
| 3 |
-
- en
|
| 4 |
pipeline_tag: object-detection
|
| 5 |
tags:
|
| 6 |
- fire
|
|
@@ -9,4 +7,34 @@ tags:
|
|
| 9 |
- pytorch
|
| 10 |
base_model:
|
| 11 |
- Ultralytics/YOLO26
|
| 12 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
|
|
|
|
|
|
| 2 |
pipeline_tag: object-detection
|
| 3 |
tags:
|
| 4 |
- fire
|
|
|
|
| 7 |
- pytorch
|
| 8 |
base_model:
|
| 9 |
- Ultralytics/YOLO26
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# Safety Detection Model
|
| 13 |
+
|
| 14 |
+
A fine-tuned YOLO model for detecting fire and smoke in images and video streams, built for real-time safety monitoring.
|
| 15 |
+
|
| 16 |
+
## Model Details
|
| 17 |
+
- **Architecture:** YOLOv26 (fine-tuned)
|
| 18 |
+
- **Framework:** PyTorch
|
| 19 |
+
- **Task:** Object Detection
|
| 20 |
+
|
| 21 |
+
## Classes
|
| 22 |
+
| ID | Label |
|
| 23 |
+
|----|-------|
|
| 24 |
+
| 0 | fire |
|
| 25 |
+
| 1 | smoke |
|
| 26 |
+
|
| 27 |
+
## Dataset
|
| 28 |
+
Fine-tuned on the [Home Fire Dataset](https://www.kaggle.com/datasets/pengbo00/home-fire-dataset) from Kaggle.
|
| 29 |
+
|
| 30 |
+
## Usage
|
| 31 |
+
```python
|
| 32 |
+
from ultralytics import YOLO
|
| 33 |
+
|
| 34 |
+
model = YOLO("path/to/model.pt")
|
| 35 |
+
results = model("image.jpg")
|
| 36 |
+
```
|
| 37 |
+
|
| 38 |
+
## Limitations
|
| 39 |
+
- Trained on home fire scenarios — performance may degrade in industrial or outdoor environments
|
| 40 |
+
- Not validated for production safety-critical systems without further testing
|