Loading
Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more

PlantVillage Image Classification Tutorial

Using AlexNet on Caffe to solve the PlantVillage image classification problem.

NOTE: This tutorial uses a "Transfer Learning" or "Fine Tuning" approach to solve
the image classification problem. This tutorial should be used to assist in development of basic ideas when it comes to approaching this and similar problems.

Transfer Learning is against the rules of the PlantVillage Classification Challenge, and this tutorial is not intended to "generate" a submission. All submissions made by using a Transfer Learning approach (as described in the tutorial, or otherwise) will be disqualified.

CrowdAI’s educational vision is to become a great open access learning resource for data analysis and machine learning. To make this happen, we are launching the CrowdAI Knowledge Base, a place where everyone in the community comes together to build high quality resources to help data scientists at all levels of expertise,
from beginners to experts.

In this tutorial we will focus on the PlantVillage Classification Challenge hosted on CrowdAI.

The goal of the challenge is to classify a set of images of plant leaves into 38 possible crop-disease pairs (see original paper here).
Here are a few examples from across all the 38 crop-disease pairs represented in the PlantVillage dataset.

PlantVillageDataset

In the following sections we will walk through the basic steps of how to get started on this problem, and similar Image Classification problems using Caffe, a very powerful and popular Deep Learning framework developed by Berkley Vision and Learning Center.

The PlantVillage Classification Challenge requires the participants to train a model by using labelled images provided in the Training Set to predict a probability distribution across all the 38 crop-disease pairs (classes) for all the images in the Test Set.