Efficient object detector via dynamic prior and dynamic feature fusion
Abstract Sparse R-CNN is a new paradigm of object detection, which predicts objects in a sparse way. However, there are some limitations in Sparse R-CNN. One is the presence of weak prior information caused by fixed learnable proposal boxes and features across different images, necessitating excessive iterations for the model to refine its predictions; the other is the inadequate exploitation of multi-scale information, leading to the sub-optimal detection performance. Thus, building upon Sparse R-CNN, we propose an efficient detector that incorporates dynamic prior and dynamic feature fusion, called $D^{2}$-Det. In particular, for the dynamic prior part, a prior information generator module dynamically generates proposal features and boxes as the dynamic prior for different images to alleviate the inference-inefficient iterative refinement process of predictions, and we further propose the class scores decoupling method to reduce the computation overhead. Furthermore, for the dynamic feature fusion part, we develop a novel lightweight multi-scale feature fusion module, which dynamically aggregates features from all layers for each proposal box, enabling adaptive feature fusion and improving detection precision by nearly 2 AP. Experiments show that $D^{2}$-Det can achieve 46.6 AP on COCO 2017 with fewer computations for the backbone ResNet50, surpassing most of the state-of-the-art detectors.
Authors
- Gaobo Yang (ORCID: https://orcid.org/0000-0003-2734-659X)
- Zhili Zhou (ORCID: https://orcid.org/0000-0002-5641-7169)
- Yuling Liu (ORCID: https://orcid.org/0000-0001-8018-8020)
- Q. M. Jonathan Wu (ORCID: https://orcid.org/0000-0002-5208-7975)
- Zihang Zhang (ORCID: https://orcid.org/0009-0008-3476-1728)
Institutions
- Hunan University (CN)
- Guangzhou University (CN)
- Institute of Art (PL)
- University of Windsor (CA)
Publication Details
- Journal
- The Computer Journal
- Published
- 2024-09-16
- DOI
- https://doi.org/10.1093/comjnl/bxae082
- Citations
- 1
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 0.25
Funders
- National Natural Science Foundation of China
- Natural Science Foundation of Fujian Province