计算机视觉
人工智能
计算机科学
目标检测
图像分辨率
图像处理
图像分割
高分辨率
模式识别(心理学)
图像(数学)
遥感
地质学
作者
Kai Liu,Zhihang Fu,Sheng Jin,Ze Chen,Fan Zhou,Rongxin Jiang,Yaowu Chen,Jieping Ye
标识
DOI:10.1109/tip.2024.3501853
摘要
Enlarging input images is a straightforward and effective approach to promote small object detection. However, simple image enlargement is significantly expensive on both computations and GPU memory. In fact, small objects are usually sparsely distributed and locally clustered. Therefore, massive feature extraction computations are wasted on the non-target background area of images. Recent works have tried to pick out target-containing regions using an extra network and perform conventional object detection, but the newly introduced computation limits their final performance. In this paper, we propose to reuse the detector's backbone to conduct feature-level object-seeking and patch-slicing, which can avoid redundant feature extraction and reduce the computation cost. Incorporating with a sparse detection head, we are able to detect small objects on high-resolution inputs (e.g., 1080P or larger) for superior performance. The resulting Efficient Small Object Detection (ESOD) approach is a generic framework, which can be applied to both CNN- and ViT-based detectors to save the computation and GPU memory costs. Extensive experiments demonstrate the efficacy and efficiency of our method. In particular, our method consistently surpasses the SOTA detectors by a large margin (e.g., 8% gains on AP) on the representative VisDrone, UAVDT, and TinyPerson datasets.
科研通智能强力驱动
Strongly Powered by AbleSci AI