We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Computer Vision and Pattern Recognition

Title: Anchor Pruning for Object Detection

Abstract: This paper proposes anchor pruning for object detection in one-stage anchor-based detectors. While pruning techniques are widely used to reduce the computational cost of convolutional neural networks, they tend to focus on optimizing the backbone networks where often most computations are. In this work we demonstrate an additional pruning technique, specifically for object detection: anchor pruning. With more efficient backbone networks and a growing trend of deploying object detectors on embedded systems where post-processing steps such as non-maximum suppression can be a bottleneck, the impact of the anchors used in the detection head is becoming increasingly more important. In this work, we show that many anchors in the object detection head can be removed without any loss in accuracy. With additional retraining, anchor pruning can even lead to improved accuracy. Extensive experiments on SSD and MS COCO show that the detection head can be made up to 44% more efficient while simultaneously increasing accuracy. Further experiments on RetinaNet and PASCAL VOC show the general effectiveness of our approach. We also introduce `overanchorized' models that can be used together with anchor pruning to eliminate hyperparameters related to the initial shape of anchors. Code and models are available at this https URL
Subjects: Computer Vision and Pattern Recognition (cs.CV)
DOI: 10.1016/j.cviu.2022.103445
Cite as: arXiv:2104.00432 [cs.CV]
  (or arXiv:2104.00432v3 [cs.CV] for this version)

Submission history

From: Maxim Bonnaerens [view email]
[v1] Thu, 1 Apr 2021 12:33:16 GMT (1320kb,D)
[v2] Tue, 20 Jul 2021 06:34:37 GMT (1086kb,D)
[v3] Wed, 1 Jun 2022 11:35:54 GMT (1189kb,D)

Link back to: arXiv, form interface, contact.