›› 2020, Vol. 32 ›› Issue (8): 1457-1465.DOI: 10.3969/j.issn.1004-1524.2020.08.16

• Biosystems Engineering • Previous Articles     Next Articles

Cotton recognition based on transfer learning

WANG Jian1,2, TIAN Guangbao1,2, ZHOU Qin1,2   

  1. 1. The State Key Laboratory of Mechanical Transmission, Chongqing University, Chongqing 400044, China;
    2. College of Mechanical Engineering, Chongqing University, Chongqing 400044, China
  • Received:2020-01-16 Online:2020-08-25 Published:2020-08-28

Abstract: An important way to improve the efficiency of the intelligent cotton picker is to realize the identification of single cotton, overlapped cottons, blocked cottons to avoid false picking and missed picking. The conventional feature extraction method is difficult to achieve satisfactory results for the identification of different types of cotton. Methods based on transfer learning and feature extraction based on transfer learning combined with extreme learning machine (ELM) was proposed. Firstly, the classification layer of AlexNet, GoogleNet and ResNet-50 models was changed and the relevant parameters were set, and the trained transfer model was used to identify the cotton validation set; secondly, the trained transfer model was used to extract the characteristics of cotton data set, and then the characteristics of the training set were used to train the ELM model to calculate the accuracy of the ELM model with different hidden layer neurons. The recognition rates of AlexNet, GoogleNet and ResNet-50 transfer models were 92.03%, 93.19% and 93.68%, respectively. The accuracy of the method of feature extraction combined with ELM was 1.97, 1.34 and 1.55 percentage points higher than that of the corresponding transfer model. The results showed that the transfer model could obtain higher accuracy for small sample cotton recognition. The method based on feature extraction and ELM could further improve the accuracy.

Key words: cotton, recognition, transfer learning, extreme learning machine, feature extraction

CLC Number: