Incremental Few-Shot Object Detection with scale- and centerness-aware weight generation
Lu Zhang,Xu Yang,Lu Qi,Shaofeng Zeng,Zhiyong Liu
DOI: https://doi.org/10.1016/j.cviu.2023.103774
IF: 4.886
2023-07-14
Computer Vision and Image Understanding
Abstract:Our work targets the task of Incremental Few-Shot Detection (iFSD), which requires quickly detecting objects of novel classes after seeing only a few training examples, while maintaining the ability to detect old classes. Different from re-training a new detector to incorporate few-shot novel categories, we tackle iFSD in a direct weight generation way. Specifically, we propose to reorganize base weights by leveraging responses of novel region features, hence transferring learned information from base weights to the novel but relevant ones. To obtain representative region features for the novel weight generation, two strategies are further presented. The scale-aware strategy respects the scale variation in object detection, and assigns each object to its corresponding pyramid to obtain the scale-specific region feature. The centerness-aware strategy reorganizes region features in the spatial range, since the localization quality of each pixel on the feature map varies. Extensive experiments are conducted on the challenging MS COCO and PASCAL VOC datasets, and our method achieves promising performances under various settings with a direct weight generation fashion. Furthermore, our generated weights can be used as good initialization for fine-tuning to induce faster convergence, thus mitigating the performance drop on old classes and improving overall results in the incremental setting.
computer science, artificial intelligence,engineering, electrical & electronic