Javvanny commited on
Commit
4fc0d37
·
verified ·
1 Parent(s): eeff0fc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +94 -3
README.md CHANGED
@@ -1,3 +1,94 @@
1
- ---
2
- license: mit
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
+ }