Rice is a staple food for a significant portion of the world's population, providing essential nutrients and serving as a versatile in-gredient in a wide range of culinary traditions. Recently, the use of deep learning has enabled automated classification of rice, im-proving accuracy and efficiency. However, classical models based on first-stage training may face difficulties in distinguishing between rice varieties with similar external characteristics, thus leading to misclassifications. Considering the transparency and feasibility of model, we selected and gradually improved pure fully connected neural network to achieve classification of rice grain. The dataset we used contains both global and domestic rice images obtained from websites and laboratories respectively. First, the training mode was changed from one-stage training to two-stage training, which significantly contributes to distinguishing two similar types of rice. Secondly, the preprocessing method was changed from random tilting to horizontal or vertical position cor-rection. After those two enhancements, the accuracy of our model increased notably from 97% to 99%. In summary, two subtle methods proposed in this study can remarkably enhance the classification ability of deep learning models in terms of the classification of rice grain.


翻译:稻米是世界大部分人口的主食,提供必需营养素,并在广泛的烹饪传统中作为多功能食材。近年来,深度学习的应用实现了稻米的自动分类,提高了准确性和效率。然而,基于单阶段训练的经典模型在区分具有相似外部特征的稻米品种时可能面临困难,从而导致误分类。考虑到模型的透明度和可行性,我们选择并逐步改进纯全连接神经网络以实现稻米籽粒的分类。我们使用的数据集包含分别从网站和实验室获取的全球及国内稻米图像。首先,将训练模式从单阶段训练改为两阶段训练,这显著有助于区分两种相似的稻米类型。其次,预处理方法从随机倾斜改为水平或垂直位置校正。经过这两项改进后,我们模型的准确率从97%显著提升至99%。总之,本研究提出的两种精细方法能显著增强深度学习模型在稻米籽粒分类方面的分类能力。

0
下载
关闭预览

相关内容

国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
46+阅读 · 2015年12月31日
VIP会员
相关基金
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
46+阅读 · 2015年12月31日
Top
微信扫码咨询专知VIP会员