DeepSetNet: Predicting sets with deep neural networks
Date
2017
Authors
Rezatofighi, S.
Kumar, V.
Milan, A.
Abbasnejad, E.
Dick, A.
Reid, I.
Editors
Advisors
Journal Title
Journal ISSN
Volume Title
Type:
Conference paper
Citation
Proceedings / IEEE International Conference on Computer Vision. IEEE International Conference on Computer Vision, 2017, vol.2017, pp.5257-5266
Statement of Responsibility
S. Hamid Rezatofighi, Vijay Kumar B G, Anton Milan, Ehsan Abbasnejad, Anthony Dick, Ian Reid
Conference Name
IEEE International Conference on Computer Vision (ICCV 2017) (22 Oct 2017 - 29 Oct 2017 : Venice, ITALY)
Abstract
This paper addresses the task of set prediction using deep learning. This is important because the output of many computer vision tasks, including image tagging and object detection, are naturally expressed as sets of entities rather than vectors. As opposed to a vector, the size of a set is not fixed in advance, and it is invariant to the ordering of entities within it. We define a likelihood for a set distribution and learn its parameters using a deep neural network. We also derive a loss for predicting a discrete distribution corresponding to set cardinality. Set prediction is demonstrated on the problem of multi-class image classification. Moreover, we show that the proposed cardinality loss can also trivially be applied to the tasks of object counting and pedestrian detection. Our approach outperforms existing methods in all three cases on standard datasets.
School/Discipline
Dissertation Note
Provenance
Description
Access Status
Rights
© 2017 IEEE