GAN-STD: small target detection based on generative adversarial network
Huilin Wang,Huaming Qian,Shuai Feng
DOI: https://doi.org/10.1007/s11554-024-01446-4
IF: 2.293
2024-03-31
Journal of Real-Time Image Processing
Abstract:With the development of convolutional neural networks, significant breakthroughs have been made in deep learning-based target detection algorithms. However, existing target detection algorithms based on convolutional neural networks need to downsample the whole image to extract deep semantic information from the image, which can lead to the loss of spatial information for small targets, impressive results on large/medium-sized targets, the results are not satisfactory for small target detection. To solve the problem of small target detection and increase the detection precision of small targets, we propose an end-to-end generative adversarial network GAN-STD for small target detection in this paper. GAN-STD makes full use of the structural correlation between targets at different scales through generative adversarial networks to enhance the similar representation of small targets in the shallow feature map and large targets in the deep feature map in the feature extraction process, and reduce the difference between the representation of small targets and large targets, thus making small targets as easy to detect as large targets. In addition, for the detector to perform better localization and classification of small targets, we back-propagate the detector losses to the generator and discriminator for end-to-end training. We merge GAN-STD onto two widely used one-stage target detectors (SSD and YOLOv4) to validate the effectiveness of our proposed GAN-STD. Extensive experiments on the widely used PASCAL VOC, MS COCO, and TT100K datasets show that the proposed GAN-STD algorithm achieves excellent results for detecting small targets.
computer science, artificial intelligence,engineering, electrical & electronic,imaging science & photographic technology