Recognition model for coated red clover seeds using YOLOv5s optimized with an attention module

Xiwen Zhang, Chuanzhong Xuan, Zhanfeng Hou

Abstract


The non-destructive recognition of coated seeds is crucial for advancing studies in coating theory. Currently, the recognition of coated seeds heavily relies on manual visual inspection and machine vision detection. However, these methods pose challenges such as high misclassification rates, low recognition efficiency, and elevated labor intensity. In response to the aforementioned challenges, this study leveraged deep learning techniques to develop a coated seed recognition model named YOLO-Coated Seeds Recognition (YOLO-CSR), aiming to address the challenges posed by coated seed recognition tasks. The experiment of this study mainly includes the following steps: First, a seed coating machine was set up to coat red clover seeds, resulting in three types of coated red clover seeds. Subsequently, by collecting images of the three types of coated seeds, a coated seed image dataset was further constructed. Then, the YOLOv5s was built, incorporating the Convolutional Block Attention Module (CBAM) into the model’s backbone to enhance its ability to learn features of coated seeds. Finally, the training results of YOLO-CSR were compared with those of other classical recognition models. The experimental results showed that YOLO-CSR achieved the best recognition performance on the self-built coated seed image dataset. The average precision (AP) for recognizing the three types of coated seeds reached 98.43%, 97.91%, and 97.26%, with a mean average precision@0.5 (mAP@0.5) of 97.87%. Compared to YOLOv5, YOLO-CSR showed a 1.18% improvement in mAP@0.5. Additionally, YOLO-CSR has a model size of only 14.9 MB, with an average recognition time (ART) of 10.1 ms and a frame per second (FPS) of 99. Experimental results prove that YOLO-CSR can accurately, efficiently, and rapidly recognize coated red clover seeds. The findings of this study provide technical support for the non-destructive recognition of spherical coated seeds.
Keywords: coated seed recognition, red clover seed, YOLO; Attention Module, CNNs
DOI: 10.25165/j.ijabe.20231606.7773

Citation: Zhang X W, Xuan C Z, Hou Z F. Recognition model for coated red clover seeds using YOLOv5s optimized with an
attention module. Int J Agric & Biol Eng, 2023; 16(6): 207–214.

Keywords


coated seed recognition, red clover seed, YOLO; Attention Module, CNNs

Full Text:

PDF

References


Zielinska M, Zapotoczny P, Białobrzewski I, Zuk-Golaszewska K, Markowski M. Engineering properties of red clover （ Trifolium pratense L.） seeds. Industrial Crops and Products, 2012; 37（1）: 69–75.

Zava D T, Dollbaum C M, Blen M. Estrogen and progestin bioactivity of foods, herbs, and spices. Experimental Biology and Medicine, 1998; 217（3）: 369–378.

Mapiye C, Aalhus J L, Turner T D, Vahmani P, Baron V S, McAllister T A, Block H C, Uttaro B, Dugan M E R. Inclusion of sunflower seed and wheat dried distillers’ grains with solubles in a red clover silage-based diet enhances steers performance, meat quality and fatty acid profiles. Animal, 2014; 8（12）: 1999–2010.

Beck V, Rohr U, Jungbauer A. Phytoestrogens derived from red clover: An alternative to estrogen replacement therapy? The Journal of Steroid Biochemistry Molecular Biology, 2005; 94（5）: 499–518.

Ma Y. Seed coating with beneficial microorganisms for precision agriculture. Biotechnology Advances, 2019; 37（7）: 107423.

Yang D B, Wang N, Yan X J, Shi J, Zhang M, Wang Z, et al. Microencapsulation of seed-coating tebuconazole and its effects on physiology and biochemistry of maize seedlings. Colloids and Surfaces B:Biointerfaces, 2014; 114: 241–246.

Ryu C M, Kima J, Choi O, Kima S H, Park C S. Improvement of biological control capacity of Paenibacillus polymyxa E681 by seed pelleting on sesame. Biological Control, 2006; 39（3）: 282–289.

Qiu Y, Amirkhani M, Mayton H, Chen Z, Taylor A G. Biostimulant seed coating treatments to improve cover crop germination and seedling growth. Agronomy, 2020; 10（2）: 154.

Hou Z F, Qiu Y, Chen Z, Liu H Y, Guo F. Mi L K. 2020. Optimization of process parameters of pelletizer for agropyron seeds under vibration force field. INMATEH-Agricultural Engineering, 2020; 60（1）: 299–308.

Zhou D Q, Fan Y L, Deng G R, He F G, Wang M L. A new design of sugarcane seed cutting systems based on machine vision. Computers and Electronics in Agriculture, 2020; 175: 105611.

Kussul N, Lavreniuk M, Skakun S, Shelestov A. Deep learning classification of land cover and crop types using remote sensing data. IEEE Geoscience and Remote Sensing Letter, 2017; 14（5）: 778–782.

Zhang J, Qu M Z, Gong Z Y, Cheng F. Online double-sided identification and eliminating system of unclosed-glumes rice seed based on machine vision. Measurement, 2022; 187: 110252.

Bai J Q, Hao F Q, Cheng F H, Li C G. Machine vision-based supplemental seeding device for plug seedling of sweet corn. Computers and Electronics Agriculture, 2021; 188: 106345.

Ansari N, Ratri S S, Jahan A, Ashik-E-Rabbani M, Rahman A. Inspection of paddy seed varietal purity using machine vision and multivariate analysis. Journal of Agriculture and Food Research, 2021; 3: 100109.

Przybylo J, Jablonski M. Using deep convolutional neural network for oak acorn viability recognition based on color images of their sections. Computers and Electronics in Agriculture, 2019; 156: 490–499.

Sau S, Ucchesu, M, Dondini L, De Franceschi P, D’hallewin G, Bacchetta G. Seed morphometry is suitable for apple-germplasm diversity-analyses. Computers and Electronics in Agriculture, 2018; 151: 118–125.

Liu W, Zhou Z Y, Xu X L, Gu Q Y, Zou S S, He W Z, et al. Evaluation method of rowing performance and its optimization for UAV-based shot seeding device on rice sowing. Computers and Electronics in Agriculture, 2023; 207: 107718.

Hu Z W, Yang H, Lou T T, Yan H W. Concurrent channel and spatial attention in Fully Convolutional Network for individual pig image segmentation. Int J Agric & Biol Eng, 2023; 16（1）: 232–242.

Zhu H F, Yang L H, Fei J W, Zhao L G, Han Z Z. Recognition of carrot appearance quality based on deep feature and support vector machine. Computers and Electronics in Agriculture, 2021; 186: 106185.

Jermsittiparsert K, Abdurrahman A, Siriattakul P, Sundeeva L A, Hashim W, Rahim R, et al. Pattern recognition and features selection for speech emotion recognition model using deep learning. International Journal of Speech Technology, 2020; 23: 799–806.

Tu K L, Wen S Z, Cheng Y, Zhang T T, Pan T, Wang J, et al. A non-destructive and highly efficient model for detecting the genuineness of maize variety ‘JINGKE 968’ using machine vision combined with deep learning. Computers and Electronics in Agriculture, 2021; 182: 106002.

Kamilaris A., Prenafeta-Boldú F X. Deep learning in agriculture: A survey. Computers and Electronics in Agriculture, 2018; 147: 70–90.

Pearline S A, Kumar V S, Harini S. A study on plant recognition using conventional image processing and deep learning approaches. Journal of Intelligent & Fuzzy Systems, 2019; 36（3）: 1997–2004.

Loddo A. , Loddo M. , Ruberto C. A novel deep learning based approach for seed image classification and retrieval. Computers and Electronics in Agriculture, 2021; 187: 106269.

Wang Y, Song S R. Variety identification of sweet maize seeds based on hyperspectral imaging combined with deep learning. Infrared Physics & Technology, 2023; 130: 104611.

Yu Z Y, Fang H, Zhangjin Q N, Mi C X, Feng X P, He Y. Hyperspectral imaging technology combined with deep learning for hybrid okra seed recognition. Biosystems Engineering, 2021; 212: 46–61.

Suo R, Gao F F, Zhou Z X, Fu L S, Song Z Z, Dhupia J, et al. Improved multi-classes kiwifruit recognition in orchard to avoid collisions during robotic picking. Computers and Electronics in Agriculture, 2021; 182: 106052.

Hou Z F, Zhang X W, Chen Z, Dai N Z, Ma X J, Liu M. Design and experiment of identification and detection system for pelleted coated seeds. Transactions of the Chinese Society for Agricultural Machinery, 2022; 53（06）: 62–69, 183.

Zhang X W, Hou Z F, Xuan C Z. Design and experiment of recognition system for coated red clover seeds based on machine vision. INMATEH-Agricultural Engineering, 2022; 66（1）: 62–72.

Zhang X W, Hou Z F, Dai N Z. Design and experiment of a new rotary coating machine based on LabView. INMATEH-Agricultural Engineering, 2021; 65（3）: 91–100.

Zhang X W, Xuan C Z, Xue J, Chen B y, Ma Y H. SR-YOLO: A high-precision, lightweight model for sheep face recognition on the mobile end. Animals, 2023; 13: 1824.

Wu D H, Lv S C, Jiang M, Song H B. Using channel pruning-based YOLO v4 deep learning algorithm for the real-time and accurate recognition of apple flowers in natural environments. Computers and Electronics in Agriculture, 2020; 178: 105742.

Han G J, He M, Zhao F, Xu Z P, Zhang M, Qin L. Insulator detection and damage identification based on improved lightweight YOLOv4 network. Energy Reports, 2021; 7（7）: 187–197.

Chen F E, Liang X M, Chen L H, Liu B Y, Lan Y B. Novel method for real-time recognition and tracking of pig body and its different parts. Int J Agric & Biol Eng, 2020; 13（6）: 144–149.

Yang B H, Gao Z W, Gao Y, Zhu Y. Rapid detection and counting of wheat ears in the field using YOLOv4 with attention module. Agronomy, 2021; 11（6）: 1202.

Xu D, Wu Y. Improved YOLO-V3 with DenseNet for multi-scale remote sensing target detection. Sensors, 2020; 20: 4276.

Zhang X W, Xuan C Z, Ma Y H, Su H, Zhang M G. Biometric facial identification using attention module optimized YOLOv4 for sheep. Computers and Electronics in Agriculture, 2022; 203: 107452.

Zhang X D, Kang X, Feng N N, Liu G. Automatic recognition of dairy cow mastitis from thermal images by a deep learning detector. Computers and Electronics in Agriculture, 2020; 178: 105754.

Wu J, Tang T, Chen M, Wang Y, Wang K. A study on adaptation lightweight architecture based deep learning models for bearing fault diagnosis under varying working conditions. Expert Systems with Applications, 2020; 160: 113710.




Copyright (c) 2023 International Journal of Agricultural and Biological Engineering

Creative Commons License
This work is licensed under a Creative Commons Attribution 4.0 International License.

2023-2026 Copyright IJABE Editing and Publishing Office