食品与机械2024，Vol.40Issue(7)：111-116,6.DOI:10.13652/j.spjx.1003.5788.2023.81032
基于视觉 Transformer 的马铃薯薯形与大小自动分级
Automatic grading of potato shape and size based on visual Transformer
摘要
Abstract
[Objective]Aiming at the problems of low accuracy and poor robustness in the previous classification methods based on artificial features.[Methods]A strong generalization automatic classification method of potato shape and size was proposed in this study.First,two potato ViT models were built based on Transformer model to complete potato shape grading and size grading tasks in parallel.Secondly,a robust model was trained by using migration strategy and data augmentation method.Finally,the effectiveness of this method in potato grading was verified by quantitative analysis of test sets.[Results]The experimental results show that the potato ViT model achieves 96.36％and 94.75％for potato shape classification,and 89.66％and 85.16％for size grading in terms of accuracy and μF1 index.The classification accuracy was better than VGG16,ResNet50 and MobileNetV3 network models.[Conclusion]The results shows that it is feasible to apply the proposed method for the real-time and accurate detection of potato shape and size.The results of this study can provide theoretical and technical support for potato intelligent grading.关键词
马铃薯/薯形/大小/分级/TransformerKey words
potatoes/shape/size/grading/Transformer引用本文复制引用
樊宏鹏,于鹏飞,杨森..基于视觉 Transformer 的马铃薯薯形与大小自动分级[J].食品与机械,2024,40(7):111-116,6.基金项目
甘肃省教育厅高校教师创新基金项目(编号:2023A-226) （编号:2023A-226）
兰州资源环境职业技术大学科研能力提升项目(编号:X2023A-11) （编号:X2023A-11）
