arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.12197cs.CV

Detective: An Attentive Recurrent Model for Sparse Object Detection

Karlsruhe Institute of Technology(卡尔斯鲁厄理工学院)

Amine Kechaou, Manuel Martinez, Monica Haurilet, Rainer Stiefelhagen

英文摘要:

In this work, we present Detective - an attentive object detector that identifies objects in images in a sequential manner. Our network is based on an encoder-decoder architecture, where the encoder is a convolutional neural network, and the decoder is a convolutional recurrent neural network coupled with an attention mechanism. At each iteration, our decoder focuses on the relevant parts of the image using an attention mechanism, and then estimates the object's class and the bounding box coordinates. Current object detection models generate dense predictions and rely on post-processing to remove duplicate predictions. Detective is a sparse object detector that generates a single bounding box per object instance. However, training a sparse object detector is challenging, as it requires the model to reason at the instance level and not just at the class and spatial levels. We propose a training mechanism based on the Hungarian algorithm and a loss that balances the localization and classification tasks. This allows Detective to achieve promising results on the PASCAL VOC object detection dataset. Our experiments demonstrate that sparse object detection is possible and has a great potential for future developments in applications where the order of the objects to be predicted is of interest.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑