A method for detecting the quality of cotton seeds based on an improved ResNet50 model

• Published in: PloS one Vol. 18; no. 2; p. e0273057
• Main Authors: Du, Xinwu, Si, Laiqiang, Li, Pengfei, Yun, Zhihao
• Format: Journal Article
• Language: English
• Published: United States Public Library of Science 15.02.2023; Public Library of Science (PLoS)
• Subjects: Accidental Injuries; Accuracy; Algorithms; Biology and Life Sciences; Classification; Computer and Information Sciences; Controlled conditions; Cotton; Deep learning; Feature extraction; Gossypium; Image enhancement; Machine learning; Mathematical optimization; Methods; Microbalances; Modelling; Neural networks; Optimization; Physical Sciences; Quality management; Recognition, Psychology; Research and Analysis Methods; Seeds; Social Sciences; Soybeans; Training; Transfer learning; Vision systems; China
• ISSN: 1932-6203; 1932-6203
• DOI: 10.1371/journal.pone.0273057

The accurate and rapid detection of cotton seed quality is crucial for safeguarding cotton cultivation. To increase the accuracy and efficiency of cotton seed detection, a deep learning model, which was called the improved ResNet50 (Impro-ResNet50), was used to detect cotton seed quality. First, the convolutional block attention module (CBAM) was embedded into the ResNet50 model to allow the model to learn both the vital channel information and spatial location information of the image, thereby enhancing the model's feature extraction capability and robustness. The model's fully connected layer was then modified to accommodate the cotton seed quality detection task. An improved LRelu-Softplus activation function was implemented to facilitate the rapid and straightforward quantification of the model training procedure. Transfer learning and the Adam optimization algorithm were used to train the model to reduce the number of parameters and accelerate the model's convergence. Finally, 4419 images of cotton seeds were collected for training models under controlled conditions. Experimental results demonstrated that the Impro-ResNet50 model could achieve an average detection accuracy of 97.23% and process a single image in 0.11s. Compared with Squeeze-and-Excitation Networks (SE) and Coordination Attention (CA), the model's feature extraction capability was superior. At the same time, compared with classical models such as AlexNet, VGG16, GoogLeNet, EfficientNet, and ResNet18, this model had superior detection accuracy and complexity balances. The results indicate that the Impro-ResNet50 model has a high detection accuracy and a short recognition time, which meet the requirements for accurate and rapid detection of cotton seed quality.