# sifat192/aircraft-damage-classification architecture

> Architecture diagram of the GitHub repository sifat192/aircraft-damage-classification, generated by GitDiagram from its file tree, README and a sample of source files (last updated 2026-09-30). It is an AI-made overview: check details against the source.

- Interactive diagram: https://gitdiagram.com/sifat192/aircraft-damage-classification
- Repository: https://github.com/sifat192/aircraft-damage-classification

## Overview

This repository is a notebook-based aircraft-damage image classifier using VGG16 transfer learning. Its documented workflow takes images through preprocessing and augmentation, trains a custom classification head over pretrained features, then evaluates and visualizes predictions. The README also documents prediction on unseen images. The notebook itself was not excerpted, so component paths point to the notebook and relationships below are supported by the README rather than inspected implementation details.

Main entry points: [`Aircraft_Damage_Classification_VGG16.ipynb`](https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb) (Model Training)

## Diagram (Mermaid)

```mermaid
flowchart TD

subgraph group_data["Data Preparation"]
  node_images[("Aircraft Images")]
  node_preprocess["Image Preprocessing"]
  node_augment["Data Augmentation"]
end

subgraph group_model["Model Training"]
  node_vgg16["VGG16 Features"]
  node_classifier["Custom Classifier"]
  node_training["Model Training"]
end

subgraph group_results["Evaluation and Prediction"]
  node_evaluation["Model Evaluation"]
  node_metrics["Accuracy and Loss"]
  node_prediction["Unseen Image Prediction"]
  node_visualization["Prediction Visualization"]
end

node_user(("User"))

node_user -->|"provides"| node_images
node_images -->|"loads and processes"| node_preprocess
node_preprocess -->|"augments training data"| node_augment
node_augment -->|"supplies images"| node_vgg16
node_vgg16 -->|"extracts features for"| node_classifier
node_classifier -->|"trains during"| node_training
node_training -->|"evaluates"| node_evaluation
node_evaluation -->|"reports"| node_metrics
node_training -->|"enables"| node_prediction
node_prediction -->|"visualizes"| node_visualization
node_user -->|"requests"| node_prediction

click node_preprocess "https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb"
click node_augment "https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb"
click node_vgg16 "https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb"
click node_classifier "https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb"
click node_training "https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb"
click node_evaluation "https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb"
click node_metrics "https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb"
click node_prediction "https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb"
click node_visualization "https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb"

classDef toneNeutral fill:#f8fafc,stroke:#334155,stroke-width:1.5px,color:#0f172a
classDef toneBlue fill:#dbeafe,stroke:#2563eb,stroke-width:1.5px,color:#172554
classDef toneAmber fill:#fef3c7,stroke:#d97706,stroke-width:1.5px,color:#78350f
classDef toneMint fill:#dcfce7,stroke:#16a34a,stroke-width:1.5px,color:#14532d
classDef toneRose fill:#ffe4e6,stroke:#e11d48,stroke-width:1.5px,color:#881337
classDef toneIndigo fill:#e0e7ff,stroke:#4f46e5,stroke-width:1.5px,color:#312e81
classDef toneTeal fill:#ccfbf1,stroke:#0f766e,stroke-width:1.5px,color:#134e4a
class node_images,node_preprocess,node_augment,node_user toneBlue
class node_vgg16,node_classifier,node_training toneAmber
class node_evaluation,node_metrics,node_prediction,node_visualization toneMint
```

## Components

### Data Preparation

- **Aircraft Images**
- [Image Preprocessing](https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb): `Aircraft_Damage_Classification_VGG16.ipynb`
- [Data Augmentation](https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb): `Aircraft_Damage_Classification_VGG16.ipynb`

### Model Training

- [VGG16 Features](https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb): `Aircraft_Damage_Classification_VGG16.ipynb`
- [Custom Classifier](https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb): `Aircraft_Damage_Classification_VGG16.ipynb`
- [Model Training](https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb): `Aircraft_Damage_Classification_VGG16.ipynb`

### Evaluation and Prediction

- [Model Evaluation](https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb): `Aircraft_Damage_Classification_VGG16.ipynb`
- [Accuracy and Loss](https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb): `Aircraft_Damage_Classification_VGG16.ipynb`
- [Unseen Image Prediction](https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb): `Aircraft_Damage_Classification_VGG16.ipynb`
- [Prediction Visualization](https://github.com/sifat192/aircraft-damage-classification/blob/main/Aircraft_Damage_Classification_VGG16.ipynb): `Aircraft_Damage_Classification_VGG16.ipynb`

### Other components

- **User**

## Connections

- User → Aircraft Images: provides (evidence: `README.md`)
- Aircraft Images → Image Preprocessing: loads and processes (evidence: `README.md`)
- Image Preprocessing → Data Augmentation: augments training data (evidence: `README.md`)
- Data Augmentation → VGG16 Features: supplies images (evidence: `README.md`)
- VGG16 Features → Custom Classifier: extracts features for (evidence: `README.md`)
- Custom Classifier → Model Training: trains during (evidence: `README.md`)
- Model Training → Model Evaluation: evaluates (evidence: `README.md`)
- Model Evaluation → Accuracy and Loss: reports (evidence: `README.md`)
- Model Training → Unseen Image Prediction: enables (evidence: `README.md`)
- Unseen Image Prediction → Prediction Visualization: visualizes (evidence: `README.md`)
- User → Unseen Image Prediction: requests (evidence: `README.md`)