Return
Using multi-label classification to improve object detection
DOI:10.1016/j.neucom.2019.08.089.png)
Abstract
En 中文
In this paper, a novel multi-task framework for object detection is proposed. The framework uses multi-label classification as an auxiliary task to improve object detection, and can be trained and tested endto-end. The object detection branch adopts R-FCN methods to solve the object detection task. The multi-label branch uses attention mechanism to solve the multi-label classification task. The features, which are generated by the attention mechanism in the multi-label branch, contain rough localization information of the objects. Thus, the features can be useful for the object detection. Both the box-level features and the image-level features of multi-label are fused to improve the accuracy of the object detection. The proposed framework does not require any extra annotation, since the ground truth of the multi-label classification can be directly obtained from the bounding box annotations. This is different from other multi-task frameworks such as StuffNet and Mask R-CNN which need extra semantic segmentation and instance segmentation annotations. Extensive experiments on PASCAL VOC 2007, PASCAL VOC 2012 and MS COCO demonstrate the effectiveness of the proposed approach. Code has been made publicly available at: https://github.com/GT9505/MONet. (C) 2019 Elsevier B.V. All rights reserved.
Keywords:
Object detection
Multi-label classification
Feature fusion
Deep learning
AI Summary
Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

