通过自适应缩放提升无人机小目标检测,无需修改主模型
Adaptive Image Zoom-in with Bounding Box Transformation for UAV Object Detection
- 设计非均匀缩放机制,动态聚焦小目标区域
- 在SeaDronesSee上使mAP提升超8.4点,延迟仅增3毫秒
- 适配任意检测架构,适合小目标密集场景应用
由于无人机图像中目标尺寸小,检测难度大。本文提出一种轻量级自适应缩放框架ZoomDet,通过非均匀缩放增强小目标特征表达。核心包括:轻量级偏移预测与基于框的缩放目标,实现高效缩放;提出角对齐框变换方法,在缩放空间中训练并反向映射预测框。在VisDrone、UAVDT和SeaDronesSee三个典型无人机检测数据集上验证,该方法与检测架构解耦,可通用。尤其在SeaDronesSee上,使用Faster R-CNN时mAP提升超过8.4个百分点,额外延迟仅约3毫秒。代码已开源。
原文摘要 · Abstract (English)
Detecting objects from UAV-captured images is challenging due to the small object size. In this work, a simple and efficient adaptive zoom-in framework is explored for object detection on UAV images. The main motivation is that the foreground objects are generally smaller and sparser than those in common scene images, which hinders the optimization of effective object detectors. We thus aim to zoom in adaptively on the objects to better capture object features for the detection task. To achieve the goal, two core designs are required: \textcolor{black}{i) How to conduct non-uniform zooming on each image efficiently? ii) How to enable object detection training and inference with the zoomed image space?} Correspondingly, a lightweight offset prediction scheme coupled with a novel box-based zooming objective is introduced to learn non-uniform zooming on the input image. Based on the learned zooming transformation, a corner-aligned bounding box transformation method is proposed. The method warps the ground-truth bounding boxes to the zoomed space to learn object detection, and warps the predicted bounding boxes back to the original space during inference. We conduct extensive experiments on three representative UAV object detection datasets, including VisDrone, UAVDT, and SeaDronesSee. The proposed ZoomDet is architecture-independent and can be applied to an arbitrary object detection architecture. Remarkably, on the SeaDronesSee dataset, ZoomDet offers more than 8.4 absolute gain of mAP with a Faster R-CNN model, with only about 3 ms additional latency. The code is available at https://github.com/twangnh/zoomdet_code.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。