Hugging Face's logo

Ashegh-Sad-Warrior
/
yolo_aerial_detection_

TensorBoard
Model card Files Metrics Community
yolo_aerial_detection_
File size: 2,690 Bytes 
3c16880
 
 
 
 
 
 
 
 
b81d753
3c16880
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
b81d753
3c16880
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
 
# README File Content for Hugging Face

# YOLO Aerial Detection Model

This repository contains a YOLO-based model for aerial object detection. The model is trained to detect and classify various objects from aerial images, such as planes, ships, vehicles, and more. Below are the details and visualizations regarding the performance of the model.

## Model Description

- **Model Name**: YOLO Aerial Mine Detection
- **Framework**: Ultralytics YOLOv11n-obb
- **Languages**: English, Persian
- **Classes Detected**:
  - Plane (هواپیما)
  - Ship (کشتی)
  - Storage Tank (مخزن ذخیره)
  - Baseball Diamond (زمین بیسبال)
  - Tennis Court (زمین تنیس)
  - Basketball Court (زمین بسکتبال)
  - Ground Track Field (زمین دو و میدانی)
  - Harbor (بندرگاه)
  - Bridge (پل)
  - Large Vehicle (خودرو بزرگ)
  - Small Vehicle (خودرو کوچک)
  - Helicopter (هلیکوپتر)
  - Roundabout (میدان)
  - Soccer Ball Field (زمین فوتبال)
  - Swimming Pool (استخر شنا)

## Training Details

- **Dataset**: Custom aerial images annotated for object detection.
- **Metrics**: Precision, Recall, mAP@0.5, F1 Score
- **Training Environment**: Kaggle, GPU-accelerated environment
- **Optimizer**: SGD
- **Libraries Used**:
  - **Ultralytics**: YOLOv11n-obb (version 8.0.0)
  - **Gradio**: For creating the user interface (version 3.1.4)
  - **Pandas**: For data handling (version 1.3.3)
  - **Pillow**: For image manipulation (version 8.4.0)
  - **OpenCV**: For video processing (version 4.5.3)

## Evaluation Results

Below are the various evaluation results obtained during the training and testing phases of the model.

### F1-Confidence Curve
![F1-Confidence Curve](F1_curve.png)

### Precision-Confidence Curve
![Precision-Confidence Curve](P_curve.png)

### Precision-Recall Curve
![Precision-Recall Curve](PR_curve.png)

### Recall-Confidence Curve
![Recall-Confidence Curve](R_curve.png)

### Confusion Matrix
![Confusion Matrix](confusion_matrix.png)

### Labels Correlogram
![Labels Correlogram](labels_correlogram.jpg)

### Labels Distribution
![Labels Distribution](labels.jpg)

## How to Use

1. Clone this repository.
2. Load the model using the Ultralytics YOLO library.
3. Use the model for object detection on aerial images.

python
from ultralytics import YOLO

# Load the trained model
yolo_model = YOLO('yolo11n-obb.pt')

# Perform detection
yolo_model('test_image.jpg')



## License
This model is open-sourced under the MIT License.

## Acknowledgements
Special thanks to the Kaggle community and Hugging Face for providing tools and platforms for developing and sharing this project.