Grape disease prediction using TensorFlow with Python

By Saket Singh

We will predict the disease of the grape plant by looking at their leaves using TensorFlow in Python, we have four classes of disease of grapes we will classify plant into infected or healthy.

We will be using Python throughout this article.

It is a classification problem and we will be using transfer learning with mobilenet as it is a very good architecture with very fewer parameters to train making it more useful for applications like embedding in a web page or mobile application.

so before starting off make sure you understand the concept of transfer learning in TensorFlow otherwise you will find few things confusing also we will be using colab notebook for training our model.

Dataset used in this article is available on kaggle, I have also attached a link of drive in text file to directly access the dataset from the packet itself.

I hope you will like and understand things here.

We will be using Tensorflow for this, to install TensorFlow:-

pip install tensorflow

 

Happy Learning