/flight_classification

Deep learning model to classify flights

Primary LanguageJupyter Notebook

Flight Classification

Deep learning model to classify flight images.

Dataset

I have prepared the following three kinds of images. Here is the dataset link

  • helicopter
  • passenger-plane
  • rocket

Deep learning model

I am using a pre-trained ResNet50 model to train the model with an above dataset. I have fine-tuned the ResNet50 model to work with our custom dataset. I have trained the model. You can download the flight classification model here

How to check model accuracy

You can checkout Flight Prediction.ipynb Colab Notebook to see the image prediction results.

Model Performance(Confusion Matrix)