e1250 commited on
Commit
3bd709e
·
verified ·
1 Parent(s): 15ba09b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -3
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