Javvanny commited on
Commit
1de59b5
·
verified ·
1 Parent(s): 040f9e5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +70 -60
README.md CHANGED
@@ -2,91 +2,101 @@
2
  license: mit
3
  pipeline_tag: object-detection
4
  ---
5
- 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.
 
6
 
7
- Model Summary
8
  This model has been trained to identify the following classes:
 
 
 
 
 
9
 
10
- Drone (UAV copter)
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 below shows the normalized detection accuracy across different classes. Key insights include:
41
 
42
- Drone Detection: 85% accuracy, with occasional misclassifications as background.
43
- Airplane Detection: Excellent accuracy of 99%.
44
- Helicopter Detection: Correctly identified 67% of the time, with some confusion with birds.
45
- Bird Detection: 68% accuracy, with some misclassifications as helicopters.
46
- Background: Some non-object areas are occasionally detected as objects.
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
- Airport Surveillance: Detecting drones and birds to prevent collisions and ensure safety.
52
- Military and Security Operations: Monitoring restricted airspaces for unauthorized drones or other aerial vehicles.
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
- bash
 
 
59
  pip install -r requirements.txt
60
- Run Inference
61
- Load the model and run inference on images or video frames using the sample inference.py script:
62
 
63
- from yolov8 import YOLO
 
64
 
 
 
65
  model = YOLO("yolov8m_fly_obj_detection.pt")
66
  results = model.predict("image.jpg")
67
- Output
 
68
  The model outputs bounding boxes for each detected object, along with their respective class labels and confidence scores.
69
 
70
- Example Results
71
- Class True Positive Rate Common Misclassifications
72
- Drone 85% Background
73
- Airplane 99% None
74
- Helicopter 67% Bird
75
- Bird 68% Helicopter
76
- Limitations
77
- Class Confusion: Some confusion exists between similar classes (e.g., helicopters and birds).
78
- Background Misclassification: Non-object areas may occasionally be misclassified as objects.
79
- License
 
 
 
 
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
+ ```