Classifying-Plants-using-Transfer-Learning
Classifying-Plants-using-Transfer-Learning copied to clipboard
Retraining Inception-v3 model to perform image segmentation using Tensorflow.
Link to the Publication:
IJARCET-VOL-5-ISSUE-11-2664-2669.pdf
Classifying Plant types using Transfer Learning
We intend to perform image segmentation on the plant phenotyping dataset. The dataset contains images(rgb coloured and masked labels) of 3 different classes of plants(1 and 2 : both consisting of top-view time-lapse images of Arabidopsis thaliana rosettes; and 3 : Tobacco plants).
The images of these 3 classes of plants are stored in the sub-folders A1, A2 and A3 inside the folder LSCData.
The accuracy obtained via transfer learning using Google's Inception-v3 model was 98%.
- Firstly, we set up docker.
- Then we install and run the Tensorflow docker image.
- We organise our dataset.
- Retrain the inception model for our own dataset.
- Test our classifier using the label_image.py script.
You can use this CodeLab by Google as a guide. Also this tutorial is quite helpful.
##Dataset
The dataset can be downloaded from here. The technical report included with the dataset describes the data acquisition, plant material, and environmental conditions in detail.
##Requirements
##Usage
-
Start the docker image
docker run -it -v ~/projects/tf_files/:/tf_files/ gcr.io/tensorflow/tensorflow:latest-devel
-
Run the label_image script to label the image.
python /tf_files/label_image.py <path_to_file>