Skip to content

Abdo-Mohammed-10/Aircraft-Damage-Classification-Captioning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

16 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

✈️ Aircraft Damage: Classification & Captioning πŸ“Έ

Open In Colab

πŸ“Œ Overview

This comprehensive Deep Learning project automates the inspection of aircraft surfaces. It not only classifies structural damage (dents vs. cracks) using Computer Vision but also generates natural language descriptions of the damage using Generative AI. The system combines a VGG16 classifier with the BLIP (Bootstrapping Language-Image Pre-training) model to provide both categorical labels and detailed textual summaries.

πŸš€ Key Features

πŸ” Damage Classification:

Uses a pre-trained VGG16 model (Transfer Learning) to accurately classify images as dent or crack.

πŸ“ Automated Captioning:

Integrates the Salesforce BLIP model to generate human-readable captions and summaries of the damage (e.g., "This is a detailed photo showing a dent on the fuselage").

πŸ”„ Hybrid Framework:

Demonstrates advanced interoperability by wrapping a PyTorch Hugging Face model within a custom TensorFlow/Keras layer (BlipCaptionSummaryLayer).

πŸ–ΌοΈ End-to-End Pipeline:

Automated dataset downloading, image preprocessing (rescaling/augmentation), training, evaluation, and inference.

πŸ“Š Visual Analytics:

Includes tools for visualizing training metrics (Loss/Accuracy) and displaying prediction grids with Ground Truth vs. Predicted labels.

πŸ› οΈ Technologies Used

Deep Learning:

              🧠 TensorFlow, Keras, PyTorch

Generative AI:

              πŸ€— Hugging Face Transformers (BLIP Model)

Computer Vision:

              πŸ‘οΈ VGG16 (ImageNet weights), OpenCV/PIL

Data Handling:

              🐼 NumPy, Pandas, Zipfile/Tarfile

Visualization:

              πŸ“‰ Matplotlib

πŸ“‚ Dataset

The system utilizes the Aircraft Damage Dataset, automatically downloaded and split into:

Training/Validation: For optimizing the VGG16 classifier.

Testing: For evaluating classification accuracy and generating caption inferences.

About

✈️ AI Aircraft Inspector: Combines VGG16 for damage classification πŸ” and BLIP for generative image captioning πŸ“. 🧠 Hybrid Architecture: A unified pipeline merging TensorFlow and PyTorch to detect and describe structural defects πŸ’₯.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors