Electronic Science and Technology ›› 2023, Vol. 36 ›› Issue (11): 19-27.doi: 10.16180/j.cnki.issn1007-7820.2023.11.004

Previous Articles     Next Articles

Scene Recognition Algorithm Based on Deep Transfer Learning and Multi-Scale Feature Fusion

WANG Qiao,HU Chunyan,LI Feifei   

  1. School of Optical-Electrical and Computer Engineering, University of Shanghai for Science and Technology, Shanghai 200093, China
  • Received:2022-05-24 Online:2023-11-15 Published:2023-11-20
  • Supported by:
    Professor of Special Appointment(EasternScholar) at Shanghai Institutions of Higher Learning(ES2015XX)

Abstract:

CNN(Convolutional Neural Networks) hase achieved excellent results in the field of scene recognition research, but this method do not fully take into account the particularity of the scene. Due to different scales, viewpoints, and backgrounds, there exists large intra-class variation within the same scene class. On the other hand, the common objects also result in a certain inter-class similarities among heterogeneous scenes as well. Considering that scene images of different scales will affect the size of objects in them, this study proposes a scene recognition algorithm based on deep transfer learning and multi-scale feature fusion. First, the network parameters pre-trained on the Places data set are migrated to the CNN model used in this study using migration learning, and then the network is fine-tuned and retrained to reduce the training cost. Secondly, the multi-scale image blocks obtained from the class activation map are fed into the CNN for feature extraction, and the obtained feature vectors are fused to make the final scene image features more comprehensive. Experiment results carried out on the SUN397 data set show that compared with other CNN-based algorithms, the proposed algorithm significantly improves the accuracy of scene recognition.

Key words: scene recognition, convolutional neural network, SE-Block, class activation map, transfer learning, multi-scale, feature fusion, support vector machine

CLC Number: 

  • TP391