4.6 Article

Oriented Vehicle Detection in Aerial Images Based on YOLOv4

Journal

SENSORS
Volume 22, Issue 21, Pages -

Publisher

MDPI
DOI: 10.3390/s22218394

Keywords

object detection; oriented bounding box; aerial image

Funding

  1. Ministry of Science and Technology, Taiwan [110-2634-F-001-007]

Ask authors/readers for more resources

This paper proposes a simple and efficient oriented object detector based on the YOLOv4 architecture. By regressing the offset of an object's front point instead of its angle or corners, the discontinuous boundary problem caused by angular periodicity or corner order is avoided. The introduction of the intersection over union (IoU) correction factor makes the training process more stable. Experimental results show that the proposed method outperforms other methods in terms of detection speed and accuracy.
CNN-based object detectors have achieved great success in recent years. The available detectors adopted horizontal bounding boxes to locate various objects. However, in some unique scenarios, objects such as buildings and vehicles in aerial images may be densely arranged and have apparent orientations. Therefore, some approaches extend the horizontal bounding box to the oriented bounding box to better extract objects, usually carried out by directly regressing the angle or corners. However, this suffers from the discontinuous boundary problem caused by angular periodicity or corner order. In this paper, we propose a simple but efficient oriented object detector based on YOLOv4 architecture. We regress the offset of an object's front point instead of its angle or corners to avoid the above mentioned problems. In addition, we introduce the intersection over union (IoU) correction factor to make the training process more stable. The experimental results on two public datasets, DOTA and HRSC2016, demonstrate that the proposed method significantly outperforms other methods in terms of detection speed while maintaining high accuracy. In DOTA, our proposed method achieved the highest mAP for the classes with prominent front-side appearances, such as small vehicles, large vehicles, and ships. The highly efficient architecture of YOLOv4 increases more than 25% detection speed compared to the other approaches.

Authors

I am an author on this paper
Click your name to claim this paper and add it to your profile.

Reviews

Primary Rating

4.6
Not enough ratings

Secondary Ratings

Novelty
-
Significance
-
Scientific rigor
-
Rate this paper

Recommended

No Data Available
No Data Available