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