Using Data-Driven Computer Vision Techniques to Improve Wheat Yield Prediction

Abstract

Accurate ear counting is essential for determining wheat yield, but traditional manual methods are labour-intensive and time-consuming. This study introduces an innovative approach by developing an automatic ear-counting system that leverages machine learning techniques applied to high-resolution images captured by unmanned aerial vehicles (UAVs). Drone-based images were captured during the late growth stage of wheat across 15 fields in Bosnia and Herzegovina. The images, processed to a resolution of 1024 × 1024 pixels, were manually annotated with regions of interest (ROIs) containing wheat ears. A dataset consisting of 556 high-resolution images was compiled, and advanced models including Faster R-CNN, YOLOv8, and RT-DETR were utilised for ear detection. The study found that although lower-quality images had a minor effect on detection accuracy, they did not significantly hinder the overall performance of the models. This research demonstrates the potential of digital technologies, particularly machine learning and UAVs, in transforming traditional agricultural practices. The novel application of automated ear counting via machine learning provides a scalable, efficient solution for yield prediction, enhancing sustainability and competitiveness in agriculture.