Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Align Deep Features for Oriented Object Detection

About

The past decade has witnessed significant progress on detecting objects in aerial images that are often distributed with large scale variations and arbitrary orientations. However most of existing methods rely on heuristically defined anchors with different scales, angles and aspect ratios and usually suffer from severe misalignment between anchor boxes and axis-aligned convolutional features, which leads to the common inconsistency between the classification score and localization accuracy. To address this issue, we propose a Single-shot Alignment Network (S$^2$A-Net) consisting of two modules: a Feature Alignment Module (FAM) and an Oriented Detection Module (ODM). The FAM can generate high-quality anchors with an Anchor Refinement Network and adaptively align the convolutional features according to the anchor boxes with a novel Alignment Convolution. The ODM first adopts active rotating filters to encode the orientation information and then produces orientation-sensitive and orientation-invariant features to alleviate the inconsistency between classification score and localization accuracy. Besides, we further explore the approach to detect objects in large-size images, which leads to a better trade-off between speed and accuracy. Extensive experiments demonstrate that our method can achieve state-of-the-art performance on two commonly used aerial objects datasets (i.e., DOTA and HRSC2016) while keeping high efficiency. The code is available at https://github.com/csuhan/s2anet.

Jiaming Han, Jian Ding, Jie Li, Gui-Song Xia• 2020

Related benchmarks

TaskDatasetResultRank
Oriented Object DetectionDOTA v1.0 (test)
SV80.94
378
Object DetectionDOTA 1.0 (test)
Plane AP89.67
256
Oriented Object DetectionDOTA (test)
AP (Plane)89.11
92
Object DetectionHRSC 2016 (test)
mAP@0.0790.17
72
Object DetectionDroneVehicle (test)
mAP5064.8
61
Oriented Object DetectionHRSC 2016 (test)--
55
Rotated Object DetectionDOTA 1.0 (test)
mAP79.42
46
Oriented Object DetectionHRSC2016--
35
Oriented Object DetectionDIOR-R (test)
APL67.98
28
Object DetectionDOTA
AP (Plane)89.28
28
Showing 10 of 28 rows

Other info

Code

Follow for update