Abstract:Aerial image objects are usually orientated arbitrarily, with a large scale range, and densely distributed. Traditional horizontal bounding box (HBB) detectors tend to filter out densely distributed objects leading to missed detections, such as ship (SH) and vehicle. Therefore, oriented object detection has become a mainstream solution in recent years. The 2-D Gaussian distribution representation of the oriented bounding boxes (OBBs) solves the problem of angular discontinuity and boundary discontinuity and thus gets more attention. However, as the aspect ratio of the object gradually decreases, its predicted angular performance continues to decrease. We find that the angular gradient of an object decreases sharply as the aspect ratio decreases, resulting in a large gradient gap between a small aspect ratio object (SARO) and a large aspect ratio object (LARO). It makes the detector prefer to ignore SARO during training, which weakens the high precision performance of SARO. We call this phenomenon shape imbalance. To solve the problem, we proposed a simple gradient rebalancing strategy named shape balance. Since the shape imbalance is only related to the aspect ratio of the object, we designed a modulation function with an inverse aspect ratio to calculate the balance coefficient. The principle of the function is that the larger the aspect ratio, the smaller the balance coefficient; the smaller the aspect ratio, the larger the balance coefficient. We aim to get the balance coefficients for objects with different aspect ratios. Location loss multiplied by a balance coefficient can directly adjust the gradient gap between objects with different aspect ratios to achieve a rebalancing effect. Extensive experiments conducted on DOTA-v1.0 dataset and DIOR-R dataset verify the effectiveness of our proposed method. Our method improves the detection performance of Gaussian location loss by an average of 2.08%/1.01%(AP75/mAP) metrics on the DOTA-v1.0 dataset and 1.17%/0.82%(AP75/mAP) improvements for DIOR-R dataset.

Rotated Object Detection via Scale-Invariant Mahalanobis Distance in Aerial Images

Learning Modulated Loss for Rotated Object Detection

RSDet++: Point-based Modulated Loss for More Accurate Rotated Object Detection

Rotating object detection in remote-sensing environment

FPDIoU Loss: A Loss Function for Efficient Bounding Box Regression of Rotated Object Detection

The KFIoU Loss for Rotated Object Detection

YOLOv7-Aerial: Rotated Object Detection of Aerial Images Based on Improved YOLOv7

Point RCNN: An Angle-Free Framework for Rotated Object Detection

Unmanned aerial vehicle (UAV) object detection algorithm based on keypoints representation and rotated distance-IoU loss

Adaptive Angle Module and Radian Regression Method for Rotated Object Detection

DARDet: A Dense Anchor-free Rotated Object Detector in Aerial Images

Rethinking Rotated Object Detection with Gaussian Wasserstein Distance Loss

Learning power Gaussian modeling loss for dense rotated object detection in remote sensing images

Learning High-Precision Bounding Box for Rotated Object Detection Via Kullback-Leibler Divergence

Segmentation Based Rotated Bounding Boxes Prediction and Image Synthesizing for Object Detection of High Resolution Aerial Images

Rotated YOLOv4 with Attention-wise Object Detectors in Aerial Images

Detecting Rotated Objects as Gaussian Distributions and its 3-D Generalization

ReDet: A Rotation-equivariant Detector for Aerial Object Detection

RiDOP: A Rotation-Invariant Detector with Simple Oriented Proposals in Remote Sensing Images

Precise Orientation Estimation for Rotated Object Detection Based on a Unit Vector Coding Approach

Rebalancing Gaussian Location Loss for High-Precision Detection on Remote Sensing Images