Update README.md
Browse files
README.md
CHANGED
|
@@ -8,45 +8,42 @@ pinned: false
|
|
| 8 |
---
|
| 9 |
|
| 10 |
|
| 11 |
-
#
|
| 12 |
|
| 13 |
## Model Description
|
| 14 |
|
| 15 |
-
This is a
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
### Intended Use
|
| 18 |
|
| 19 |
-
- **Primary intended uses**:
|
| 20 |
-
- **Primary intended users**:
|
| 21 |
-
- **Out-of-scope use cases**: Not intended for production use or real-world
|
| 22 |
|
| 23 |
## Training Data
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
- Split: 80% train, 20% test
|
| 28 |
-
- 8 categories of climate disinformation claims
|
| 29 |
-
|
| 30 |
-
### Labels
|
| 31 |
-
0. No relevant claim detected
|
| 32 |
-
1. Global warming is not happening
|
| 33 |
-
2. Not caused by humans
|
| 34 |
-
3. Not bad or beneficial
|
| 35 |
-
4. Solutions harmful/unnecessary
|
| 36 |
-
5. Science is unreliable
|
| 37 |
-
6. Proponents are biased
|
| 38 |
-
7. Fossil fuels are needed
|
| 39 |
|
| 40 |
## Performance
|
| 41 |
|
| 42 |
-
### Metrics
|
| 43 |
-
- **Accuracy**: ~
|
| 44 |
-
- **Environmental Impact**:
|
| 45 |
-
-
|
| 46 |
-
-
|
|
|
|
| 47 |
|
| 48 |
### Model Architecture
|
| 49 |
-
The model implements a
|
|
|
|
| 50 |
|
| 51 |
## Environmental Impact
|
| 52 |
|
|
@@ -57,15 +54,10 @@ Environmental impact is tracked using CodeCarbon, measuring:
|
|
| 57 |
This tracking helps establish a baseline for the environmental impact of model deployment and inference.
|
| 58 |
|
| 59 |
## Limitations
|
| 60 |
-
-
|
| 61 |
-
- No learning or pattern recognition
|
| 62 |
-
- No consideration of input text
|
| 63 |
-
- Serves only as a baseline reference
|
| 64 |
-
- Not suitable for any real-world applications
|
| 65 |
|
| 66 |
## Ethical Considerations
|
| 67 |
|
| 68 |
-
-
|
| 69 |
-
- Model makes random predictions and should not be used for actual classification
|
| 70 |
- Environmental impact is tracked to promote awareness of AI's carbon footprint
|
| 71 |
-
```
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
|
| 11 |
+
# Object Detection Model for Smoke detection
|
| 12 |
|
| 13 |
## Model Description
|
| 14 |
|
| 15 |
+
This is a YOLO11s model developed for the Frugal AI Challenge 2025, specifically for the object detection task of identifying smoke.
|
| 16 |
+
Training was conducted over 100 epochs using custom hyperparameters and augmented data.
|
| 17 |
+
We employed a post-training optimization phase that included an optimized engine (TensorRT), pruning, and quantization techniques to compress the model and reduce its memory footprint.
|
| 18 |
+
The objective was to achieve the best possible accuracy while minimizing energy consumption.
|
| 19 |
|
| 20 |
### Intended Use
|
| 21 |
|
| 22 |
+
- **Primary intended uses**: Detect smoke based on images from watchtowers cameras.
|
| 23 |
+
- **Primary intended users**: Firefighters to tackle early wildfires developments
|
| 24 |
+
- **Out-of-scope use cases**: Not intended for production use or real-world object detection tasks
|
| 25 |
|
| 26 |
## Training Data
|
| 27 |
+
The model uses the Pyro-SDIS is a dataset designed for wildfire smoke detection using AI models.
|
| 28 |
+
It is developed in collaboration with the Fire and Rescue Services (SDIS) in France and the dedicated volunteers of the Pyronear association.
|
| 29 |
+
- Size : Training set : 29537 - Validation set : 4099
|
| 30 |
+
- 1 category of objects to detect : Smoke
|
| 31 |
|
| 32 |
+
### Objects
|
| 33 |
+
0. Smoke
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
|
| 35 |
## Performance
|
| 36 |
|
| 37 |
+
### Metrics on Validation Dataset
|
| 38 |
+
- **Accuracy**: ~80%%
|
| 39 |
+
- **Inference Environmental Impact**:
|
| 40 |
+
- for 4099 images on a Tesla T4 GPU
|
| 41 |
+
- 1.3 gCO2 // Emissions tracked in gCO2eq
|
| 42 |
+
- 3.5 Wh // Energy consumption tracked in Wh
|
| 43 |
|
| 44 |
### Model Architecture
|
| 45 |
+
The model implements a YOLO11s with augmented data using image processing techniques such as zooming, adding noise and contrast adjustment.
|
| 46 |
+
The model was then optimized for high performance inference using TensorRT, pruning and quantization technics to compress and limit model memory footprint.
|
| 47 |
|
| 48 |
## Environmental Impact
|
| 49 |
|
|
|
|
| 54 |
This tracking helps establish a baseline for the environmental impact of model deployment and inference.
|
| 55 |
|
| 56 |
## Limitations
|
| 57 |
+
- Not suitable for industrialization or real-world applications
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
|
| 59 |
## Ethical Considerations
|
| 60 |
|
| 61 |
+
- Model makes predictions and should be used with caution, can be false predictions.
|
|
|
|
| 62 |
- Environmental impact is tracked to promote awareness of AI's carbon footprint
|
| 63 |
+
```
|