Binary-Class Image Classification Model for Military vs. Non-Military Vehicles Using TensorFlow Take 4

SUMMARY: This project aims to construct a predictive model using a TensorFlow convolutional neural network (CNN) and document the end-to-end steps using a template. The Normal vs. Military Vehicles dataset is a binary classification situation where we attempt to predict one of two possible outcomes.

INTRODUCTION: The dataset contains over 17,000 images of military and non-military vehicles. The researcher collected these images to investigate the machine learning model’s ability to understand and distinguish the basic features of military and civilian vehicles.

ANALYSIS: The VGG19 model’s performance achieved an accuracy score of 97.65% after five epochs using the validation dataset. When we applied the model to the test dataset, the model achieved an accuracy score of 97.27%.

CONCLUSION: In this iteration, the TensorFlow VGG19 CNN model appeared suitable for modeling this dataset.

Dataset ML Model: Binary classification with numerical features

Dataset Used: Normal vs. Military Vehicles

Dataset Reference: https://www.kaggle.com/datasets/amanrajbose/normal-vs-military-vehicles

One source of potential performance benchmarks: https://www.kaggle.com/datasets/amanrajbose/normal-vs-military-vehicles/code

The HTML formatted report can be found here on GitHub. [https://github.com/daines-analytics/deep-learning-projects/tree/master/py_tensorflow_binaryclass_amanrajbose_normal_military_vehicles]