Computer Vision-Aided Intelligent Monitoring of Coffee: Towards Sustainable Coffee Production

TL;DR

This study develops an AI-powered mobile application using YOLOv7 and semi-supervised learning to remotely monitor coffee plant health, estimate yield and quality, and predict harvest time, enhancing precision agriculture practices.

Contribution

It introduces a semi-supervised annotation method with K-means for coffee image analysis, surpassing supervised methods in accuracy and efficiency, and integrates this into a practical mobile app for field use.

Findings

Mean average precision of 0.89 for the model

Semi-supervised annotation achieved 0.77 precision

Mobile app enables real-time fruit ripening tracking

Abstract

Coffee which is prepared from the grinded roasted seeds of harvested coffee cherries, is one of the most consumed beverage and traded commodity, globally. To manually monitor the coffee field regularly, and inform about plant and soil health, as well as estimate yield and harvesting time, is labor-intensive, time-consuming and error-prone. Some recent studies have developed sensors for estimating coffee yield at the time of harvest, however a more inclusive and applicable technology to remotely monitor multiple parameters of the field and estimate coffee yield and quality even at pre-harvest stage, was missing. Following precision agriculture approach, we employed machine learning algorithm YOLO, for image processing of coffee plant. In this study, the latest version of the state-of-the-art algorithm YOLOv7 was trained with 324 annotated images followed by its evaluation with 82 unannotated images as test data. Next, as an innovative approach for annotating the training data, we trained K-means models which led to machine-generated color classes of coffee fruit and could thus characterize the informed objects in the image. Finally, we attempted to develop an AI-based handy mobile application which would not only efficiently predict harvest time, estimate coffee yield and quality, but also inform about plant health. Resultantly, the developed model efficiently analyzed the test data with a mean average precision of 0.89. Strikingly, our innovative semi-supervised method with an mean average precision of 0.77 for multi-class mode surpassed the supervised method with mean average precision of only 0.60, leading to faster and more accurate annotation. The mobile application we designed based on the developed code, was named CoffeApp, which possesses multiple features of analyzing fruit from the image taken by phone camera with in field and can thus track fruit ripening in real time.