Assessing Pineapple Maturity in Complex Scenarios Using an Improved Retinanet Algorithm

Abstract

In China, low levels of accuracy in predicting when pineapple crops will reach maturity can result from environmental variation such as light changes, fruit overlap, and shading. Therefore, this study proposed the use of an improved RetinaNet algorithm (ECA-Retinanet) based on the ECA attention mechanism. The ECA attention mechanism was embedded into the classification subnet of RetinaNet to improve accuracy in detecting different levels of maturity in pineapples. A new pineapple dataset was collected comprising four different growth stages under mild and severe complex scenarios. The experimental results have shown that the mAP (Mean Average Precision) and F1 score (Balanced Score) of the ECA-Retinanet model were 97.69%, 94.75%, 93.2%, and 90% for identification in mild and severe complex scenarios. These values are 0.42%, 2%, 1.78%, and 1.5% higher than the original RetinaNet model which exceeds those of the six existing state-of-the-art detection models. The results have indicated that the proposed algorithm could be used for accurate identification of pineapple fruit and can detect fruit maturity using ground color images in the natural environment. The study findings provide a technical reference for automatic picking robots and early yield estimation.