Article thumbnail

Object Detection Network Based on Feature Fusion and Attention Mechanism

By Ying Zhang, Yimin Chen, Chen Huang and Mingke Gao

Abstract

In recent years, almost all of the current top-performing object detection networks use CNN (convolutional neural networks) features. State-of-the-art object detection networks depend on CNN features. In this work, we add feature fusion in the object detection network to obtain a better CNN feature, which incorporates well deep, but semantic, and shallow, but high-resolution, CNN features, thus improving the performance of a small object. Also, the attention mechanism was applied to our object detection network, AF R-CNN (attention mechanism and convolution feature fusion based object detection), to enhance the impact of significant features and weaken background interference. Our AF R-CNN is a single end to end network. We choose the pre-trained network, VGG-16, to extract CNN features. Our detection network is trained on the dataset, PASCAL VOC 2007 and 2012. Empirical evaluation of the PASCAL VOC 2007 dataset demonstrates the effectiveness and improvement of our approach. Our AF R-CNN achieves an object detection accuracy of 75.9% on PASCAL VOC 2007, six points higher than Faster R-CNN

Topics: CNN, object detection network, attention mechanism, feature fusion, Information technology, T58.5-58.64
Publisher: MDPI AG
Year: 2019
DOI identifier: 10.3390/fi11010009
OAI identifier: oai:doaj.org/article:867de8ef6bc04552a6fba24bb883de90
Journal:
Download PDF:
Sorry, we are unable to provide the full text but you may find it at the following location(s):
  • https://doaj.org/article/867de... (external link)
  • https://doaj.org/toc/1999-5903 (external link)
  • http://www.mdpi.com/1999-5903/... (external link)
  • Suggested articles


    To submit an update or takedown request for this paper, please submit an Update/Correction/Removal Request.