arXiv:2608.28161cs.CVcs.LG2026-08中稿 · Journal of Banglad…

用深度学习帮农民识别孟加拉芒果品种,准确率超97%。

Empowering Local Agriculture: A Deep Learning-Powered Web System for Identifying Bangladeshi Mango Varieties

论文配图:Empowering Local Agriculture: A Deep Learning-Powered Web System for Identifying Bangladeshi Mango Varieties
图 1 · 摘自论文原文
  • 用EfficientNetB0模型结合图像增强,实现芒果品种自动识别
  • 测试集准确率达97.36%,多品种识别F1分数均超0.93
  • 轻量模型部署为网页工具,适合农户和农业工作者使用

孟加拉芒果品种识别困难,因近缘品种外观相似且拍摄条件多样。本文构建了一个基于深度学习的网页系统,用于自动识别孟加拉芒果品种。研究收集了2013张高质量芒果图像(3024x4032像素),分为九类,将Bari-4与Bari-7合并为一个Bari类别。数据集按70%/15%/15%划分训练、验证和测试集,并采用图像增强提升泛化能力。在三种预训练CNN架构(ResNet18、ResNet50、EfficientNetB0)上进行微调,结果表明EfficientNetB0表现最佳,验证准确率为98.01%,测试准确率为97.36%;而ResNet18和ResNet50的测试准确率分别为86.47%和78.55%。EfficientNetB0的类别级F1分数介于0.93至0.99之间,其中Bari类达到0.97。该模型约含400万参数,适合轻量化部署。研究将模型集成至Streamlit网页应用,用户可上传芒果图像,获得预测品种及概率。系统提供了一种便捷实用的识别工具,展示了深度学习在孟加拉农业中的应用潜力。

原文摘要 · Abstract (English)

Mango variety identification in Bangladesh is challenging because closely related cultivars can have similar visual characteristics and images are often captured under varying real-world conditions. This work presents a deep learning-based web system for automatic identification of Bangladeshi mango varieties. We collected 2,013 high-quality mango images (3024x4032 pixels) from local markets and farms and organized them into nine classes, combining Bari-4 and Bari-7 as a single Bari class. The dataset was divided into training (70%), validation (15%), and test (15%) sets, with image augmentation applied to improve model generalization. Three pretrained CNN architectures, ResNet18, ResNet50, and EfficientNetB0, were fine-tuned under consistent training settings. EfficientNetB0 achieved the best performance, obtaining 98.01% validation accuracy and 97.36% test accuracy, compared with 86.47% and 78.55% test accuracy for ResNet18 and ResNet50, respectively. Class-wise F1-scores for EfficientNetB0 ranged from 0.93 to 0.99, while the Bari class achieved an F1-score of 0.97. The selected EfficientNetB0 model has approximately 4 million parameters, making it suitable for lightweight deployment. We integrated the model into a Streamlit web application that enables users to upload a mango image and receive a predicted variety with class probabilities. The system provides an accessible, practical tool for mango identification and demonstrates the potential of deep learning for supporting agricultural applications in Bangladesh.

图像识别农业AI轻量模型芒果识别

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。