Instance Segmentation and Classification of Coffee Leaf Plant using Mask RCNN and Transfer Learning

Document Type : Original Article

Authors

1 Fayoum University Faculty of Engineering Electronics and Communication Engineering Dept.

2 Fayoum University, Faculty of Engineering, Electronics and Communication Eng. Dept.

Abstract

Coffee is one of the most consumed beverages in the world and is crucial in the economy of many developing countries. The effectiveness and fast classification of healthy coffee leaves are the most decisive factors in determining their quality for consumers and industrial companies. To separate coffee leaves from the complex real-world background, we proposed an instance segmentation algorithm based on transfer learning and state-of-the-art deep learning algorithm mask regional convolutional neural network (Mask RCNN) for this work. Robusta coffee leaf images dataset (RoCole) is considered in this study. In addition, the VGG Image annotator (VIA) has been used to manually annotate the dataset for coffee leaf segmentation tasks.
Resnet101 was adopted as a backbone network, combined with the Feature Pyramid Network (FPN) architecture for feature extraction. The Region Proposal Network (RPN) was trained to create region proposals for each feature map, which used to separate the input image from the complex background. The output image is then fed to a transfer learning-based binary classifier to be classified into one of the two classes. Results reveal that the proposed system has a high-test accuracy of 97.76% for the binary classifier. If the image is classified as unhealthy, it then passes through another segmentation stage based on the HSV color model to highlight the defected areas of the coffee leaf. Instance segmentation results of 148 test images showed that the mean average detection precision rate (mAP@50:95) was 100%, the mean average recall rate (recall@50:95) was 84.5%, and the F1-score was 91.6%.

Keywords

Main Subjects