Optimizing tomato plant phenotyping detection: Boosting YOLOv8 architecture to tackle data complexity

Effective identification of tomato plant traits is crucial for timely monitoring and evaluating their growth and harvest. However, conducting stress experiments on multiple tomato genotypes introduces challenges due to the nature of the data. One of these challenges arises from an imbalanced sample distribution, potentially leading to misclassification between classes and disruptions in model recognition. This paper addresses the effect of these challenges by considering the imbalanced classes of flowers, fruits, and nodes and proposing an improved detection approach through data balancing. A novel data-balancing approach is introduced in this study to overcome the issue of imbalanced data. The proposed solution involves the implementation of a YOLOv8 deep learning model, which effectively detects flowers, fruits, and nodes in tomato plants. This model significantly enhances the ability of the algorithm to detect objects of varying sizes within complex environments. To further bolster the recognition capability of the targeted classes, the proposed model integrates a Squeeze-and-Excitation (SE) block attention module into its head architecture. This module strengthens the model recognition ability by giving increased attention to the studied classes, thereby enhancing overall detection performance. The results demonstrate that the data balancing approach successfully improves the model performance in response to the data challenges. When applying the technique of pre-training the optimal weights obtained from balanced data on imbalanced data, the SE-block module showed significant improvements in outcomes.