Image Classification of Pests with Residual Neural Network Based on Transfer Learning

• Published in: Applied sciences Vol. 12; no. 9; p. 4356
• Main Authors: Li, Chen, Zhen, Tong, Li, Zhihui
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 01.05.2022
• Subjects: Accuracy; agriculture; Classification; convolutional neural network; Crops; Datasets; Deep learning; image processing; insect pest; Machine learning; Neural networks; pest recognition; transfer learning
• ISSN: 2076-3417; 2076-3417
• DOI: 10.3390/app12094356

Agriculture is regarded as one of the key food sources for humans throughout history. In some countries, more than 90% of the population lives on agriculture. However, pests are regarded as one of the major causes of crop loss worldwide. Accurate and automated technology to classify pests can help pest detection with great significance for early preventive measures. This paper proposes the solution of a residual convolutional neural network for pest identification based on transfer learning. The IP102 agricultural pest image dataset was adopted as the experimental dataset to achieve data augmentation through random cropping, color transformation, CutMix and other operations. The processing technology can bring strong robustness to the affecting factors such as shooting angles, light and color changes. The experiment in this study compared the ResNeXt-50 (32 × 4d) model in terms of classification accuracy with different combinations of learning rate, transfer learning and data augmentation. In addition, the experiment compared the effects of data enhancement on the classification performance of different samples. The results show that the model classification effect based on transfer learning is generally superior to that based on new learning. Compared with new learning, transfer learning can greatly improve the model recognition ability and significantly reduce the training time to achieve the same classification accuracy. It is also very important to choose the appropriate data augmentation technology to improve classification accuracy. The accuracy rate of classification can reach 86.95% based on the combination of transfer learning + fine-tuning and CutMix. Compared to the original model, the accuracy of classification of some smaller samples was significantly improved. Compared with the relevant studies based on the same dataset, the method in this paper can achieve higher classification accuracy for more effective application in the field of pest classification.