HCP: A Flexible CNN Framework for Multi-Label Image Classification
National University of Singapore · Beijing Jiaotong University · +1 more institution
Abstract
Convolutional Neural Network (CNN) has demonstrated promising performance in single-label image classification tasks. However, how CNN best copes with multi-label images still remains an open problem, mainly due to the complex underlying object layouts and insufficient multi-label training images. In this work, we propose a flexible deep CNN infrastructure, called Hypotheses-CNN-Pooling (HCP), where an arbitrary number of object segment hypotheses are taken as the inputs, then a shared CNN is connected with each hypothesis, and finally the CNN output results from different hypotheses are aggregated with max pooling to produce the ultimate multi-label predictions. Some unique characteristics of this flexible…
Citation impact
- FWCI
- 44.55
- Percentile
- 100%
- References
- 71
Authors
8Topics & keywords
- Computer science
- Artificial intelligence
- Pattern recognition (psychology)
- Multi-label classification
- Image (mathematics)
- Contextual image classification
- Convolutional neural network
- Computer vision
- Industry, innovation and infrastructure