Scholarlink

<p>Accurate estimation of mango size during harvesting is essential for optimizing harvest timing, resource allocation, quality control, and profitability in mango cultivation. This paper presents a computer vision based on-tree mango detection using an upgraded MangoYOLO5 model and size estimation from the detected bounding boxes using XGBoost regression analysis. As the accuracy of size estimation depends on the precision of the rendered bounding box, we have focused on improving the detection accuracy by adding spatial and channel attention modules into the MangoYOLO5, a customized YOLOv5 model for mango detection proposed in our previous work. The proposed Attention Enhanced MangoYOLO5 (AE-MangoYOLO5) model focuses on the most relevant parts of the input image, reducing false positives and improving detection accuracy in scenarios involving occlusion and overlapping. The detected bounding box parameters, such as height and width, are correlated with the actual mango dimensions measured using a Vernier Caliper. XGBoost regression with positional loss is used to get the correlation, and an empirical study is conducted by comparing its results with traditional XGBoost, linear, and logistic regression models. A coefficient of determination (<InlineEquation ID="IEq1"> <InlineMediaObject> <ImageObject Color="BlackWhite" FileRef="11694_2025_3114_Article_IEq1.gif" Format="GIF" Height="16" Rendition="HTML" Resolution="72" Type="Linedraw" Width="21" /> </InlineMediaObject> <EquationSource Format="TEX">\(R^2\)</EquationSource> <EquationSource Format="MATHML"><math> <msup> <mi>R</mi> <mn>2</mn> </msup> </math></EquationSource> </InlineEquation>) of 0.91 is obtained in the proposed approach, reflecting a robust alignment between estimated and actual mango sizes which is higher than that of the conventional regression models.</p>

错误：搜索内容不能为空，请输入英文关键词
错误：关键词超出字数限制，请精简
高级检索

On-tree mango detection and size estimation using attention-enhanced mangoYOLO5 and XGBoost regression

  • 期刊论文

  • 发表日期：

    2025年1月30日

查看全文
  • Hari Chandana Pichhika,
  • Priyambada Subudhi,
  • Raja Vara Prasad Yerra

摘要

Accurate estimation of mango size during harvesting is essential for optimizing harvest timing, resource allocation, quality control, and profitability in mango cultivation. This paper presents a computer vision based on-tree mango detection using an upgraded MangoYOLO5 model and size estimation from the detected bounding boxes using XGBoost regression analysis. As the accuracy of size estimation depends on the precision of the rendered bounding box, we have focused on improving the detection accuracy by adding spatial and channel attention modules into the MangoYOLO5, a customized YOLOv5 model for mango detection proposed in our previous work. The proposed Attention Enhanced MangoYOLO5 (AE-MangoYOLO5) model focuses on the most relevant parts of the input image, reducing false positives and improving detection accuracy in scenarios involving occlusion and overlapping. The detected bounding box parameters, such as height and width, are correlated with the actual mango dimensions measured using a Vernier Caliper. XGBoost regression with positional loss is used to get the correlation, and an empirical study is conducted by comparing its results with traditional XGBoost, linear, and logistic regression models. A coefficient of determination ( \(R^2\) R 2 ) of 0.91 is obtained in the proposed approach, reflecting a robust alignment between estimated and actual mango sizes which is higher than that of the conventional regression models.