Few-shot object detection via metric learning

Min Zhu; Chongyang Zhang

doi:10.1117/12.2622909

4 March 2022 Few-shot object detection via metric learning

Min Zhu, Chongyang Zhang

Proceedings Volume 12084, Fourteenth International Conference on Machine Vision (ICMV 2021); 120840Y (2022) https://doi.org/10.1117/12.2622909
Event: Fourteenth International Conference on Machine Vision (ICMV 2021), 2021, Rome, Italy

Abstract

To achieve good results with the existing target detection framework, a large amount of annotated data is often needed. However, the acquisition of annotated data is a laborious process. It is even impossible to obtain sufficient annotated data in some categories. To reduce the dependence of deep learning model on large-scale data, a new end-to-end single-stage detector (FSSSD) network is proposed based on metric learning method. In this way, the objects that are not seen during training can be detected under the condition of providing a small number of samples. The main innovation of this paper is that the traditional single-stage detection model FCOS is improved and the Class Feature Extractor module is added to make it become the Correlation Detector. Thus, the model can extract the feature distribution of the support category from the small number of pictures provided by the support set. Thereafter, the model converts the query set feature map into the object probability distribution map, and fuses it with the original feature map to enhance the feature representation of the potential objects consistent with the supporting category on the feature map, so that the model pays more attention to the objects consistent with the supporting category in classification and regression. The method in this paper does not need fine-tuning or retraining at all when recognizing objects of a new category, and only needs to provide supporting pictures of the corresponding category during testing. At the same time, our modules are flexible and easy to migrate, theoretically suitable for all target detection models, and can improve the performance of these models on few-shot problems.

Citation Download Citation

Min Zhu and Chongyang Zhang "Few-shot object detection via metric learning", Proc. SPIE 12084, Fourteenth International Conference on Machine Vision (ICMV 2021), 120840Y (4 March 2022); https://doi.org/10.1117/12.2622909

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
8 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Data modeling

Systems modeling

Target detection

Convolution

Detection and tracking algorithms

Prototyping

Statistical modeling

Keywords/Phrases

Search In:

Publication Years