Binary Class Image Classification Deep Learning Model for CycleGAN Van Gogh vs. Photo Using TensorFlow Take 1

Template Credit: Adapted from a template made available by Dr. Jason Brownlee of Machine Learning Mastery.

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 CycleGAN Cezanne vs. Photo dataset is a binary classification situation where we attempt to predict one of two possible outcomes.

INTRODUCTION: The CycleGAN dataset collection contains images from two classes A and B (for example, apple vs. orange, horses vs. zebras, and so on). The researchers used the images to train machine learning models for research work in General Adversarial Networks (GAN).

In this iteration, we will construct a CNN model based on the InceptionV3 architecture to make predictions.

ANALYSIS: In this iteration, the InceptionV3 model’s performance achieved an accuracy score of 99.50% after ten epochs using the training dataset. The same model processed the validation dataset with an accuracy measurement of 99.25%. Finally, the final model processed the test dataset with an accuracy score of 92.96%.

CONCLUSION: In this iteration, the InceptionV3-based CNN model appeared to be suitable for modeling this dataset. We should consider experimenting with TensorFlow for further modeling.

Dataset Used: CycleGAN Van Gogh vs. Photo Dataset

Dataset ML Model: Binary classification with numerical attributes

Dataset Reference:

One potential source of performance benchmarks: or

The HTML formatted report can be found here on GitHub.