Update README.md
Browse files
README.md
CHANGED
@@ -2,91 +2,101 @@
|
|
2 |
license: mit
|
3 |
pipeline_tag: object-detection
|
4 |
---
|
5 |
-
yolov8m_flying_objects_detection
|
|
|
6 |
|
7 |
-
Model Summary
|
8 |
This model has been trained to identify the following classes:
|
|
|
|
|
|
|
|
|
|
|
9 |
|
10 |
-
|
11 |
-
Airplane
|
12 |
-
Helicopter
|
13 |
-
Bird
|
14 |
-
Background (no object)
|
15 |
-
Classes and Objects
|
16 |
The model has been trained to detect and classify the following types of flying objects:
|
17 |
|
18 |
-
1. Drones
|
19 |
-
DJI Matrice 200
|
20 |
-
DJI Phantom 2
|
21 |
-
DJI Phantom 3
|
22 |
-
Shahed
|
23 |
-
2. Airplanes
|
24 |
-
Airbus A220
|
25 |
-
Airbus A220 (with stowed landing gear)
|
26 |
-
Airbus A380
|
27 |
-
Boeing 787
|
28 |
-
Boeing 787 (with stowed landing gear)
|
29 |
-
3. Helicopters
|
30 |
-
Bell 407
|
31 |
-
Robinson R44
|
32 |
-
4. Birds
|
33 |
-
Chayka (Seagull)
|
34 |
-
Golub (Pigeon)
|
35 |
-
5. Background
|
36 |
-
Areas with no relevant objects.
|
|
|
37 |
This breakdown provides more specific information on each class, helping users understand the diversity of objects the model can detect.
|
38 |
|
39 |
-
Confusion Matrix Analysis
|
40 |
-
The confusion matrix
|
41 |
|
42 |
-
Drone Detection
|
43 |
-
Airplane Detection
|
44 |
-
Helicopter Detection
|
45 |
-
Bird Detection
|
46 |
-
Background
|
47 |
|
48 |
-
Applications
|
49 |
This model is particularly useful in scenarios where real-time identification of airborne objects is essential. Potential applications include:
|
|
|
|
|
|
|
50 |
|
51 |
-
|
52 |
-
|
53 |
-
Wildlife Monitoring: Identifying bird movements to support ecological studies and prevent hazards.
|
54 |
-
Model Usage
|
55 |
-
Install Dependencies
|
56 |
-
Install the required packages listed in requirements.txt:
|
57 |
|
58 |
-
|
|
|
|
|
59 |
pip install -r requirements.txt
|
60 |
-
|
61 |
-
Load the model and run inference on images or video frames using the sample inference.py script:
|
62 |
|
63 |
-
|
|
|
64 |
|
|
|
|
|
65 |
model = YOLO("yolov8m_fly_obj_detection.pt")
|
66 |
results = model.predict("image.jpg")
|
67 |
-
|
|
|
68 |
The model outputs bounding boxes for each detected object, along with their respective class labels and confidence scores.
|
69 |
|
70 |
-
Example Results
|
71 |
-
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
|
|
|
|
|
|
|
|
|
80 |
This model is released under the MIT License. Feel free to use, modify, and distribute it, but please provide proper attribution.
|
81 |
|
82 |
-
Citation
|
83 |
If you use this model in your work, please consider citing it as follows:
|
84 |
|
85 |
-
bibtex
|
86 |
-
Копировать код
|
87 |
@model{yolov8m_flying_objects_detection,
|
88 |
title={YOLOv8m Flying Object Detection},
|
89 |
author={Javvanny},
|
90 |
year={2024},
|
91 |
howpublished={\url{https://huggingface.co/Javvanny/yolov8m_flying_objects_detection}},
|
92 |
-
}
|
|
|
|
2 |
license: mit
|
3 |
pipeline_tag: object-detection
|
4 |
---
|
5 |
+
# yolov8m_flying_objects_detection
|
6 |
+
`yolov8m_flying_objects_detection` is a deep learning model designed to detect various flying objects, including drones, airplanes, helicopters, and birds. Based on the YOLOv8 architecture, this model provides a strong balance of speed and accuracy, making it suitable for real-time aerial surveillance and monitoring applications.
|
7 |
|
8 |
+
## Model Summary
|
9 |
This model has been trained to identify the following classes:
|
10 |
+
- `Drone` (UAV copter)
|
11 |
+
- `Airplane`
|
12 |
+
- `Helicopter`
|
13 |
+
- `Bird`
|
14 |
+
- `Background` (no object)
|
15 |
|
16 |
+
## Classes and Objects
|
|
|
|
|
|
|
|
|
|
|
17 |
The model has been trained to detect and classify the following types of flying objects:
|
18 |
|
19 |
+
1. **Drones**
|
20 |
+
- DJI Matrice 200
|
21 |
+
- DJI Phantom 2
|
22 |
+
- DJI Phantom 3
|
23 |
+
- Shahed
|
24 |
+
2. **Airplanes**
|
25 |
+
- Airbus A220
|
26 |
+
- Airbus A220 (with stowed landing gear)
|
27 |
+
- Airbus A380
|
28 |
+
- Boeing 787
|
29 |
+
- Boeing 787 (with stowed landing gear)
|
30 |
+
3. **Helicopters**
|
31 |
+
- Bell 407
|
32 |
+
- Robinson R44
|
33 |
+
4. **Birds**
|
34 |
+
- Chayka (Seagull)
|
35 |
+
- Golub (Pigeon)
|
36 |
+
5. **Background**
|
37 |
+
- Areas with no relevant objects.
|
38 |
+
|
39 |
This breakdown provides more specific information on each class, helping users understand the diversity of objects the model can detect.
|
40 |
|
41 |
+
## Confusion Matrix Analysis
|
42 |
+
The confusion matrix above shows the normalized detection accuracy across different classes. Key insights include:
|
43 |
|
44 |
+
- **Drone Detection**: 85% accurate, with occasional misclassifications as background.
|
45 |
+
- **Airplane Detection**: Excellent accuracy of 99%.
|
46 |
+
- **Helicopter Detection**: Correctly identified 67% of the time, with some confusion with birds.
|
47 |
+
- **Bird Detection**: 68% accurate, with some misclassifications as helicopters.
|
48 |
+
- **Background**: Some non-object areas are occasionally detected as objects.
|
49 |
|
50 |
+
## Applications
|
51 |
This model is particularly useful in scenarios where real-time identification of airborne objects is essential. Potential applications include:
|
52 |
+
- **Airport Surveillance**: Detecting drones and birds to prevent collisions and ensure safety.
|
53 |
+
- **Military and Security Operations**: Monitoring restricted airspaces for unauthorized drones or other aerial vehicles.
|
54 |
+
- **Wildlife Monitoring**: Identifying bird movements to support ecological studies and prevent hazards.
|
55 |
|
56 |
+
## Model Usage
|
57 |
+
To use the model, follow these steps:
|
|
|
|
|
|
|
|
|
58 |
|
59 |
+
### Install Dependencies
|
60 |
+
Install the required packages listed in `requirements.txt`:
|
61 |
+
```bash
|
62 |
pip install -r requirements.txt
|
63 |
+
```
|
|
|
64 |
|
65 |
+
## Run Inference
|
66 |
+
Load the model and run inference on images or video frames using the sample `inference.py` script:
|
67 |
|
68 |
+
```python
|
69 |
+
from yolov8 import YOLO
|
70 |
model = YOLO("yolov8m_fly_obj_detection.pt")
|
71 |
results = model.predict("image.jpg")
|
72 |
+
```
|
73 |
+
|
74 |
The model outputs bounding boxes for each detected object, along with their respective class labels and confidence scores.
|
75 |
|
76 |
+
## Example Results
|
77 |
+
|
78 |
+
| Class | True Positive Rate | Common Misclassifications |
|
79 |
+
|------------|--------------------|---------------------------|
|
80 |
+
| Drone | 85% | Background |
|
81 |
+
| Airplane | 99% | None |
|
82 |
+
| Helicopter | 67% | Bird |
|
83 |
+
| Bird | 68% | Helicopter |
|
84 |
+
|
85 |
+
## Limitations
|
86 |
+
- **Class Confusion**: Some confusion exists between similar classes (e.g., helicopters and birds).
|
87 |
+
- **Background Misclassification**: Non-object areas may occasionally be misclassified as objects.
|
88 |
+
|
89 |
+
## License
|
90 |
This model is released under the MIT License. Feel free to use, modify, and distribute it, but please provide proper attribution.
|
91 |
|
92 |
+
## Citation
|
93 |
If you use this model in your work, please consider citing it as follows:
|
94 |
|
95 |
+
```bibtex
|
|
|
96 |
@model{yolov8m_flying_objects_detection,
|
97 |
title={YOLOv8m Flying Object Detection},
|
98 |
author={Javvanny},
|
99 |
year={2024},
|
100 |
howpublished={\url{https://huggingface.co/Javvanny/yolov8m_flying_objects_detection}},
|
101 |
+
}
|
102 |
+
```
|