Research Paper:
Real-Time Wheat Unsound Kernel Classification Detection Based on Improved YOLOv5
Zhaohui Zhang*1,*2,, Zengyang Zuo*2, Zhi Li*3, Yuguo Yin*4, Yan Chen*1, Tianyao Zhang*1, and Xiaoyan Zhao*1,*2
*1School of Automation and Electrical Engineering, University of Science and Technology Beijing
30 Xueyuan Road, Haidian District, Beijing 100083, China
*2Shunde Innovation School, University of Science and Technology Beijing
2 Zhihui Road, Daliang, Shunde District, Fo Shan, Guangdong 528399, China
*3College of Information Science and Engineering, Henan University of Technology
100 Lianhua Road, Zhengzhou High-Tech Development Zone, Zhengzhou, Henan 450001, China
*4Shandong Start Measurement and Control Equipment Co., Ltd.
600 Xinyi Road, Weifang Economic Development Zone, Weifang, Shandong 261101, China
Corresponding author
China is one of the largest wheat production countries in the world. The wheat quality determines the price and many other aspects. The detection methods of wheat quality mainly depend on manual labor. It costs high amount of manpower and time, and the classification results are partly affected by different individuals. With the development of machine vision, an automatic classification system was presented in this study. A wheat unsound kernel identification method based on the improved YOLOv5 algorithm was designed by adding efficient channel attention (ECA). Compared with convolutional block attention module (CBAM) and squeeze-and-excitation network (SENet), the improved YOLOv5 algorithm was selected to fit the model better. The recognition results showed that YOLOv5 with the addition of the attention mechanism had a significant improvement in average accuracy over that without it. The most significant improvement was observed with the addition of ECA-YOLOv5, with an average accuracy of 96.24%, a 10% improvement over the other two models, and a 13% improvement over the original YOLOv5. This satisfied the application requirements for detection of wheat unsound kernel.
- [1] H. Gao, T. Zhen, and Z. Li, “Detection of Wheat Unsound Kernels Based on Improved ResNet,” IEEE Access, Vol.10, pp. 20092-20101, 2022. https://doi.org/10.1109/ACCESS.2022.3147838
- [2] L. Jena, S. K. Behera, and P. K. Sethy, “Identification of Wheat Grain Using Geometrical Feature and Machine Learning,” 2021 2nd Int. Conf. for Emerging Technology (INCET), 2021. https://doi.org/10.1109/INCET51464.2021.9456281
- [3] S. Guan et al., “Ceramic Ring Defect Detection Based on Improved YOLOv5,” 2022 3rd Int. Conf. on Computer Vision, Image and Deep Learning & Int. Conf. on Computer Engineering and Applications (CVIDL & ICCEA), pp. 115-118, 2022. https://doi.org/10.1109/CVIDLICCEA56201.2022.9824099
- [4] F. Han and J. Li, “Wheat Heads Detection via YOLOv5 with Weighted Coordinate Attention,” 2022 7th Int. Conf. on Cloud Computing and Big Data Analytics (ICCCBDA), pp. 300-306, 2022. https://doi.org/10.1109/ICCCBDA55098.2022.9778925
- [5] C. H. R. Mendigoria et al., “Varietal Classification of Lactuca sativa Seeds Using an Adaptive Neuro-Fuzzy Inference System Based on Morphological Phenes,” J. Adv. Comput. Intell. Intell. Inform., Vol.25, No.5, pp. 618-624, 2021. https://doi.org/10.20965/jaciii.2021.p0618
- [6] F. Cheng, F. N. Chen, and Y. B. Ying, “Image Recognition of Unsound Wheat Using Artificial Neural Network,” 2010 2nd WRI Global Congress on Intelligent Systems, pp. 172-175, 2010. https://doi.org/10.1109/GCIS.2010.220
- [7] A. Sharma, T. Singh, and N. Garg, “Combining Near-Infrared Hyperspectral Imaging and ANN for Varietal Classification of Wheat Seeds,” 2022 3rd Int. Conf. on Intelligent Computing Instrumentation and Control Technologies (ICICICT), pp. 1103-1108, 2022. https://doi.org/10.1109/ICICICT54557.2022.9917725
- [8] V. K. Kolur and M. S. Padagatti, “Quality Identification and Grading of Wheat Grains Using Image Processing Techniques,” 2021 5th Int. Conf. on Electrical, Electronics, Communication, Computer Technologies and Optimization Techniques (ICEECCOT), pp. 497-500, 2021. https://doi.org/10.1109/ICEECCOT52851.2021.9708017
- [9] A. G. Singh, K. Singh, and S. A. Sampson, “Wheat Head Detection and Crop Health Classification System,” 2021 Innovations in Power and Advanced Computing Technologies (i-PACT), 2021. https://doi.org/10.1109/i-PACT52855.2021.9696823
- [10] X. Xia et al., “The Analysis of Wheat Appearance Quality Based on Digital Image Processing,” 2010 the 2nd Conf. on Environmental Science and Information Application Technology, pp. 231-234, 2010. https://doi.org/10.1109/ESIAT.2010.5568375
- [11] T. Georgieva, S. Penchev, and P. Daskalov, “Analysis of the Possibilities for Using Computer Vision and Spectral Analysis to Assess the Wheat Crops Condition,” 2022 8th Int. Conf. on Energy Efficiency and Agricultural Engineering (EE&AE), 2022. https://doi.org/10.1109/EEAE53789.2022.9831308
- [12] X. Zhao et al., “Cross-View Gait Recognition Based on Dual-Stream Network,” J. Adv. Comput. Intell. Intell. Inform., Vol.25, No.5, pp. 671-678, 2021. https://doi.org/10.20965/jaciii.2021.p0671
- [13] M. Genaev, S. Ekaterina, and D. Afonnikov, “Application of Neural Networks to Image Recognition of Wheat Rust Diseases,” 2020 Cognitive Sciences, Genomics and Bioinformatics (CSGB), pp. 40-42, 2020. https://doi.org/10.1109/CSGB51356.2020.9214703
- [14] J. Redmon et al., “You Only Look Once: Unified, Real-Time Object Detection,” 2016 IEEE Conf. on Computer Vision and Pattern Recognition (CVPR), pp. 779-788, 2016. https://doi.org/10.1109/CVPR.2016.91
- [15] R. Fan and Z. Qiu, “Improved YOLOv5 Algorithm Based on CBAM Attention Mechanism,” 2022 Int. Conf. on Frontiers of Artificial Intelligence and Machine Learning (FAIML), pp. 229-233, 2022. https://doi.org/10.1109/FAIML57028.2022.00051
This article is published under a Creative Commons Attribution-NoDerivatives 4.0 Internationa License.