HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Small Object Detection for Birds with Swin Transformer

{Ichiro Ide Takahiro Komamizu Takatsugu Hirayama Yasutomo Kawanishi TingWei Liu Marc A. Kastner Da Huo}

Abstract

Object detection is the task of detecting objects in an image. In this task, the detection of small objects is particularly difficult. Other than the small size, it is also accompanied by difficulties due to blur, occlusion, and so on. Current small object detection methods are tailored to small and dense situations, such as pedestrians in a crowd or far objects in remote sensing scenarios. However, when the target object is small and sparse, there is a lack of objects available for training, making it more difficult to learn effective features. In this paper, we propose a specialized method for detecting a specific category of small objects; birds. Particularly, we improve the features learned by the neck; the sub-network between the backbone and the prediction head, to learn more effective features with a hierarchical design. We employ Swin Transformer to upsample the image features. Moreover, we change the shifted window size for adapting to small objects. Experiments show that the proposed Swin Transformer-based neck combined with CenterNet can lead to good performance by changing the window sizes. We further find that smaller window sizes (default 2) benefit mAPs for small object detection.

Benchmarks

BenchmarkMethodologyMetrics
small-object-detection-on-sod4sb-private-testSwin Transformer + Hierarchical design
AP50: 22.6
small-object-detection-on-sod4sb-public-test-1Swin Transformer + Hierarchical design
AP50: 70.2

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Small Object Detection for Birds with Swin Transformer | Papers | HyperAI