1. A Fast and Accurate Few-Shot Detector for Objects with Fewer Pixels in Drone Image
- Author
-
Yuanlong Hou, Qiang Gao, Yuxuan Gao, and Hou Runmin
- Subjects
Similarity (geometry) ,Computer Networks and Communications ,Computer science ,lcsh:TK7800-8360 ,02 engineering and technology ,010501 environmental sciences ,drone ,01 natural sciences ,Discriminative model ,Bounding overwatch ,0202 electrical engineering, electronic engineering, information engineering ,Siamese framework ,Computer vision ,Electrical and Electronic Engineering ,0105 earth and related environmental sciences ,Pixel ,business.industry ,Perspective (graphical) ,lcsh:Electronics ,few-shot ,object detection ,Object detection ,Drone ,Hardware and Architecture ,Control and Systems Engineering ,Feature (computer vision) ,Signal Processing ,small object ,020201 artificial intelligence & image processing ,Artificial intelligence ,business ,attention mechanism ,anchor-free - Abstract
Unmanned aerial vehicles (UAVs) are important in modern war, and object detection performance influences the development of related intelligent drone application. At present, the target categories of UAV detection tasks are diversified. However, the lack of training samples of novel categories will have a bad impact on the task. At the same time, many state-of-the-arts are not suitable for drone images due to the particularity of perspective and large number of small targets. In this paper, we design a fast few-shot detector for drone targets. It adopts the idea of anchor-free in fully convolutional one-stage object detection (FCOS), which leads to a more reasonable definition of positive and negative samples and faster speed, and introduces Siamese framework with more discriminative target model and attention mechanism to integrate similarity measures, which enables our model to match the objects of the same categories and distinguish the different class objects and background. We propose a matching score map to utilize the similarity information of attention feature map. Finally, through soft-NMS, the predicted detection bounding boxes for support category objects are generated. We construct a DAN dataset as a collection of DOTA and NWPU VHR-10. Compared with many state-of-the-arts on the DAN dataset, our model is proved to outperform them for few-shot detection tasks of drone images.
- Published
- 2021