Skip to content

Tactical Command Dashboard for Military Aircraft Detection using Deep Learning (CNN, MobileNetV2, EfficientNetB0) featuring real-time video inference, HUD simulation, and comprehensive model performance analytics built with Streamlit.

License

Notifications You must be signed in to change notification settings

RazerArdi/Military-Aircraft-Detection

Repository files navigation

βš”οΈ ARS: AERIAL RECONNAISSANCE SYSTEM

Comparative Framework: Custom CNN vs. MobileNetV2 vs. EfficientNetB0

Status Framework Technique Dataset Language

Live Deployment β€’ Methodology β€’ Benchmarks β€’ Assessment


Hero Image

πŸ“– Project Ontology

Aerial Reconnaissance System (ARS) is a computer vision framework engineered to classify military aircraft assets from aerial and ground imagery. In the domain of national defense and airspace surveillance, the ability to rapidly and accurately identify assets is paramount.

This project implements a multi-architecture approach to benchmark the performance of a Custom Convolutional Neural Network (CNN) built from scratch against state-of-the-art Transfer Learning architectures (MobileNetV2 and EfficientNetB0). The study focuses on finding the optimal trade-off between classification accuracy, model size, and inference latency for deployment on tactical interfaces.

πŸ“˜ Dataset: Military Aircraft Detection (96 Classes)

This dataset is built for fine-grained object detection of military aircraft. It covers 96 aircraft types, with some variants merged due to very similar airframes or external features.

Category Aircraft Types
Attack & Bomber A-10, B-1, B-2, B-21, B-52, F-117, F-14, F-15, F-16, F-2, F-22, F-35, F-4, F/A-18, F-CK-1, J-10, J-20, J-35, J-36, J-50, JAS-39, JF-17, JH-7, KAAN, KF-21, Mirage2000, Rafale, SR-71, Su-24, Su-25, Su-34, Su-47, Su-57, Tejas, Tornado, Tu-160, Tu-22M, Tu-95, U-2, Vulcan, X-29, X-32, XB-70, XQ-58, YF-23
Transport & Cargo A-400M, AG-600, An-124, An-22, An-225, An-72, Be-200, C-1, C-130, C-17, C-2, C-390, C-5, CL-415, Il-76, KC-135, P-3, US-2, Y-20
Helicopters AH-64, CH-47, CH-53, Ka-27, Ka-52, Mi-24, Mi-26, Mi-28, Mi-8, UH-60, WZ-10, WZ-9, Z-10, Z-19
UAVs AKINCI, MQ-25, MQ-9, RQ-4, TB-001, TB-2, WZ-7
AEW&C / Special Mission E-2, E-7, KJ-600
Trainer & Light Attack EMB-314

πŸ“Š Summary

  • Total classes: 96
  • Main groups: Attack/Bomber, Transport, Helicopter, UAV, AEW&C, Trainer

πŸ›  Methodology

The system adheres to a rigorous End-to-End Machine Learning pipeline, visualized below:

graph LR
    A[Data Ingestion] --> B(Preprocessing)
    B --> C{Model Selection}
    C -->|Scratch| D[Custom CNN]
    C -->|Transfer Learning| E[MobileNetV2]
    C -->|Transfer Learning| F[EfficientNetB0]
    D --> G[Evaluation]
    E --> G
    F --> G
    G --> H[Streamlit Dashboard]

Loading
  1. Data Ingestion: Aggregation of military aircraft imagery (Fighter jets, Bombers, UAVs, Transports).

  2. Preprocessing: Resizing (224x224), Normalization, and Label Encoding.

  3. Architecture Design:

    • Custom CNN: A heavy 5-layer convolutional network for baseline establishment.
    • MobileNetV2: Optimized for speed and low latency (Edge-ready).
    • EfficientNetB0: Designed for scaling and feature extraction depth.
  4. Deployment: Integration into a tactical dashboard using Streamlit with HUD Simulation.


βœ… UMM Laboratory Assessment

This project is submitted to fulfill the Final Practicum Assignment (UAP) for the Informatics Laboratory at UMM.

No Requirement Component Project Specification & Implementation Status
1. Pemilihan Topik Data Citra (Image Data). Topik: Klasifikasi Pesawat Militer (Aerial Reconnaissance). βœ…
2. Pengumpulan Dataset Dataset berjumlah > 15,000 gambar. Sumber: Kaggle Military Aircraft Dataset & Open Defense Repo. βœ…
3. Implementasi Model - Base: Custom CNN (Non-Pretrained)
- Pretrained 1: MobileNetV2 (Transfer Learning)
- Pretrained 2: EfficientNetB0 (Transfer Learning)
βœ…
4. Evaluasi & Analisis Evaluasi mencakup Accuracy, F1-Score, Grafik Loss, dan Confusion Matrix. (Lihat bagian Benchmarks) βœ…
5. Sistem Website Streamlit Web App (Local & Cloud).
Fitur: Input Gambar/Video, Real-time Inference, Tactical HUD.
πŸ”— Live Demo
βœ…
6. Dokumentasi Repository GitHub terstruktur dengan source code, .ipynb, dataset, dan dokumentasi lengkap. βœ…

πŸ“Š Comparative Results

The following data is extracted directly from the evaluation pipeline.

1. Performance & Efficiency Summary

Architecture Accuracy F1-Score (Weighted) Inference Time (ms) Model Size (MB)
MobileNetV2 (TL) 44.10% 0.4203 5.31 ms 24.67 MB
Custom CNN (Base) 38.19% 0.3714 4.99 ms 299.14 MB
EfficientNetB0 (TL) 4.93% 0.0049 5.38 ms 39.70 MB

πŸ§ͺ Analysis:

  • MobileNetV2 is the optimal choice, achieving the highest accuracy and F1-score with a very compact model size (24 MB), making it suitable for edge deployment.
  • Custom CNN shows extremely high storage consumption (~300 MB) despite being a simple architecture, due to the large number of parameters in Dense layers.
  • EfficientNetB0 experienced convergence failure (Underfitting) in this specific training run, likely due to hyperparameter sensitivity or dataset noise.

2. Hardest Classes to Detect

Classes with 0.0 F1-Score, indicating high confusion or insufficient training samples:

Model Top 3 Hardest Classes
Custom CNN Su47, FCK1, XQ58
MobileNetV2 MQ25, J50, CH53
EfficientNetB0 A10, Su34, Su25

πŸ’» Interface & Deployment

The system features a Tactical Command Dashboard designed for clarity and situational awareness.

Analytics Dashboard Live Inference (Tactical HUD)
Provides comprehensive multi-dimensional comparison using Radar Charts and performance metrics. Real-time object classification supporting Image and Video inputs with OpenCV HUD overlay.
Analytics Dashboard Live Inference HUD

πŸ“‰ Deep Dive Analysis

Training Dynamics (Loss & Accuracy)

Visualizing the learning stability over epochs across all three architectures.

Custom CNN MobileNetV2 EfficientNetB0
Custom CNN History MobileNetV2 History EfficientNetB0 History
Moderate convergence with signs of overfitting.
Overfitting
Stable convergence and best generalization.
Good fit
Failed to converge (flatline), indicating underfitting.
Underfitting

Confusion Matrix

Visualizing misclassifications across all three models. MobileNetV2 shows the strongest diagonal (Correct Predictions).

Custom CNN MobileNetV2 EfficientNetB0
Figure: Confusion Matrix (Custom CNN). Figure: Confusion Matrix (MobileNetV2). Figure: Confusion Matrix (EfficientNetB0).

πŸš€ Installation & Setup

To run the Tactical Dashboard locally on your machine:

# 1. Clone the repository
git clone [https://github.com/RazerArdi/Military-Aircraft-Detection](https://github.com/RazerArdi/Military-Aircraft-Detection)

# 2. Navigate to directory
cd ...

# 3. Install dependencies
pip install -r requirements.txt

# 4. Run the application
streamlit run Interface/streamlit_app.py

Directory Structure:

ARS-Command-Center/
β”œβ”€β”€ Interface/          # Streamlit Frontend (app.py)
β”œβ”€β”€ Models/             # Trained .h5 files
β”œβ”€β”€ Reports/            # Evaluation CSVs & Graphs
β”œβ”€β”€ Notebook/           # Jupyter Notebooks (.ipynb)
β”œβ”€β”€ requirements.txt    # Python Dependencies
└── README.md           # Documentation

Acknowledgements

I would like to express my gratitude to the following resources and communities that made this project possible.

πŸ“š Dataset & Resources

This project utilizes the Military Aircraft Detection Dataset, which was instrumental in training and evaluating the deep learning models.

πŸ› οΈ Frameworks & Libraries

This project relies on the open-source ecosystem:

  • TensorFlow & Keras: For model architecture and training.
  • Streamlit: For building the interactive Tactical Dashboard.
  • OpenCV: For image processing and HUD visualization.
  • Plotly: For interactive data visualization.

πŸŽ“ Academic Context

  • Institution: Universitas Muhammadiyah Malang (UMM)
  • Department: Informatics Engineering
  • Course: Machine Learning (Final Practicum Assignment)

πŸ“ Citation

If you use this code, data analysis, or the Tactical Dashboard in your research or project, please cite this repository as follows:

APA Format

Bayu Ardiyansyah. (2025). Military Aircraft Detection: Comparative Framework of CNN, MobileNetV2, and EfficientNetB0 [Source code]. GitHub. https://github.com/RazerArdi/Military-Aircraft-Detection

BibTeX Format

@software{Ardiyansyah_Military_Aircraft_Detection_2025,
  author = {Bayu Ardiyansyah},
  month = {12},
  title = {{Military Aircraft Detection: A Comparative Framework}},
  url = {[https://github.com/RazerArdi/Military-Aircraft-Detection](https://github.com/RazerArdi/Military-Aircraft-Detection)},
  version = {1.0.0},
  year = {2025},
  publisher = {GitHub}
}

Developed by Bayu Ardiyansyah

Informatics Department β€’ Universitas Muhammadiyah Malang

Β© 2025 ARS Project

About

Tactical Command Dashboard for Military Aircraft Detection using Deep Learning (CNN, MobileNetV2, EfficientNetB0) featuring real-time video inference, HUD simulation, and comprehensive model performance analytics built with Streamlit.

Topics

Resources

License

Contributing

Stars

Watchers

Forks

Packages

No packages published

Languages