高级检索+

基于深度残差网络与迁移学习的水稻虫害图像识别

Image recognition of rice diseases and insect pests based on deep residual network and transfer learning

  • 摘要: 提出一个针对大多数类型的水稻害虫的图像识别方法。对ResNet34网络进行改进,提高网络的识别能,以实现基于给定的图像自动地识别分类出主要害虫。此外,基于迁移学习方法有效避免由于数据量缺乏而使得训练不足的问题。通过ImageNet数据库开展网络参数预训练能够进一步提升网络的提取性能,通过IDADP数据库可以开展参数微调工作以及训练工作。将提出的改进ResNet34模型与其他模型的性能进行对比评估。结果显示,改进ResNet34模型的识别准确度最高,F1-score达到0.98,证明所提模型对水稻病虫害图像具有较好的识别效果。

     

    Abstract: This paper presents an image recognition method for most types of rice pests. ResNet34 network is improved to realize the recognition ability of the network, so as to automatically recognize and classify the main pests based on the given image. In addition, the transfer learning method overcomes the shortcomings of insufficient training caused by insufficient data. Firstly, the ImageNet database is used for network parameter pre training to make the network have good feature extraction ability. According to the migration learning method, the IDADP database is used for parameter fine-tuning and training. In this paper, the performance of the proposed improved ResNet34 model is compared with other models. The results show that the improved ResNet34 model has the highest recognition accuracy, and the F1-score is 0.98, which proves that the proposed model has a good recognition effect on rice pest images.

     

/

返回文章
返回