Improving the Detection of Small Oriented Objects in Aerial Images
arXiv:2401.06503 · doi:10.1109/WACVW58289.2023.00023
Abstract
Small oriented objects that represent tiny pixel-area in large-scale aerial images are difficult to detect due to their size and orientation. Existing oriented aerial detectors have shown promising results but are mainly focused on orientation modeling with less regard to the size of the objects. In this work, we proposed a method to accurately detect small oriented objects in aerial images by enhancing the classification and regression tasks of the oriented object detection model. We designed the Attention-Points Network consisting of two losses: Guided-Attention Loss (GALoss) and Box-Points Loss (BPLoss). GALoss uses an instance segmentation mask as ground-truth to learn the attention features needed to improve the detection of small objects. These attention features are then used to predict box points for BPLoss, which determines the points' position relative to the target oriented bounding box. Experimental results show the effectiveness of our Attention-Points Network on a standard oriented aerial dataset with small object instances (DOTA-v1.5) and on a maritime-related dataset (HRSC2016). The code is publicly available.
arXiv commentC. T. C. Doloriel and R. D. Cajote, "Improving the Detection of Small Oriented Objects in Aerial Images," 2023 IEEE/CVF Winter Conference on Applications of Computer Vision Workshops (WACVW), Waikoloa, HI, USA, 2023, pp. 176-185, doi: 10.1109/WACVW58289.2023.00023
References in corpus (10)
- An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale
- Neural Machine Translation by Jointly Learning to Align and Translate
- Arbitrary-Oriented Scene Text Detection via Rotation Proposals
- Gliding vertex on the horizontal bounding box for multi-oriented object detection
- R2CNN: Rotational Region CNN for Orientation Robust Scene Text Detection
- Mask R-CNN
- Learning High-Precision Bounding Box for Rotated Object Detection via Kullback-Leibler Divergence
- Rethinking Rotated Object Detection with Gaussian Wasserstein Distance Loss
- The KFIoU Loss for Rotated Object Detection
- AlphaRotate: A Rotation Detection Benchmark using TensorFlow