Semi-Supervised Person Detection in Aerial Images with Instance Segmentation and Maximum Mean Discrepancy Distance

• Published in: Remote sensing (Basel, Switzerland) Vol. 15; no. 11; p. 2928
• Main Authors: Zhang, Xiangqing, Feng, Yan, Zhang, Shun, Wang, Nan, Mei, Shaohui, He, Mingyi
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 04.06.2023
• Subjects: Ablation; Accuracy; aerial-based person detection; Algorithms; Boxes; copy-paste mechanism; data collection; Datasets; Detectors; Drones; Entropy; Image processing; Image resolution; Image segmentation; Instance segmentation; Labels; Learning; maximum mean discrepancy; Methods; Object recognition; prediction; prototypes; Ratios; Remote sensing; Search and rescue; Search and rescue operations; semi-supervised object detection; Sensors; System effectiveness; Unmanned aerial vehicles
• ISSN: 2072-4292; 2072-4292
• DOI: 10.3390/rs15112928

Detecting sparse, small, lost persons with only a few pixels in high-resolution aerial images was, is, and remains an important and difficult mission, in which a vital role is played by accurate monitoring and intelligent co-rescuing for the search and rescue (SaR) system. However, many problems have not been effectively solved in existing remote-vision-based SaR systems, such as the shortage of person samples in SaR scenarios and the low tolerance of small objects for bounding boxes. To address these issues, a copy-paste mechanism (ISCP) with semi-supervised object detection (SSOD) via instance segmentation and maximum mean discrepancy distance is proposed (MMD), which can provide highly robust, multi-task, and efficient aerial-based person detection for the prototype SaR system. Specifically, numerous pseudo-labels are obtained by accurately segmenting the instances of synthetic ISCP samples to obtain their boundaries. The SSOD trainer then uses soft weights to balance the prediction entropy of the loss function between the ground truth and unreliable labels. Moreover, a novel evaluation metric MMD for anchor-based detectors is proposed to elegantly compute the IoU of the bounding boxes. Extensive experiments and ablation studies on Heridal and optimized public datasets demonstrate that our approach is effective and achieves state-of-the-art person detection performance in aerial images.