Deep learning model to classify flight images.

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

• helicopter
• passenger-plane
• rocket

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

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