Computer and Modernization ›› 2021, Vol. 0 ›› Issue (04): 61-67.

Previous Articles     Next Articles

Flower Recognition Based on ResNet and Attention Mechanism

  

  1. (College of Computer and Information, Hohai University, Nanjing 211100, China)
  • Online:2021-04-22 Published:2021-04-25

Abstract: Flower recognition has important application value in life, and the traditional flower recognition methods have some problems, such as low recognition accuracy and weak generalization ability. To solve these problems, this paper proposes a ResNet34 network model with attention mechanism. After the first convolutional layer and each residual block of ResNet34, channel attention mechanism and spatial attention mechanism are added, and the transfer learning is used for training network model. Experiment shows that ResNet34 has a higher recognition accuracy rate than AlexNet, VGG-16 and GoogLeNet on the flower data set. The ResNet34 model with attention mechanism and transfer learning has 6.1 percentage points higher recognition accuracy than the original model, and 1.1 percentage points higher recognition accuracy than the original model with transfer learning only. Compared with traditional deep learning models, the model proposed in this paper significantly improves the recognition accuracy.

Key words: deep learning, ResNet34, attention mechanism, transfer learning, flower recognition