Direction Prediction Redefinition: Transfer Angle to Scale in Oriented Object Detection
Oriented object detection has garnered significant attention. However, rotational symmetry and discontinuity at boundaries can confuse networks, leading to discontinuous loss and regression inconsistency. In this paper, we propose an efficient multi-directional object detection framework named Direction Prediction Redefinition (DPR). We describe the angle variation of rotated bounding boxes (<inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"> <tex-math notation="LaTeX">$B_{r}$ </tex-math></inline-formula>) as changes in the dimensions of horizontal bounding boxes (<inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"> <tex-math notation="LaTeX">$B_{h}$ </tex-math></inline-formula>). Specifically, we generate two sets of horizontal bounding boxes by predicting the center points of the corresponding boundaries within the rotated bounding box, thereby avoiding boundary issues caused by angle prediction. To further achieve robust rotated boundary representation, we propose the Joint Scale Representation method and the State Feature Encoding module, which are used to eliminate outliers in rotated boundaries and guide the correct selection of horizontal bounding box vertices, respectively. Moreover, we further abstract DPR as Multiple Trigonometric functions based DPR (DPR-MT). This method maps a single angle into four sets of trigonometric functions and considers them as the four sides of the horizontal bounding box. This approach predicts angles in the form of horizontal bounding boxes without complex operations, making it plug-and-play. Experimental results and visual analysis on challenging datasets further verify the effectiveness and competitiveness of our proposed method.
Authors
- Jia Xin Wu (ORCID: https://orcid.org/0000-0002-1371-5801)
- Beihang Song (ORCID: https://orcid.org/0000-0003-2179-4949)
- Jing Li (ORCID: https://orcid.org/0000-0002-8181-0886)
- Jun Wan (ORCID: https://orcid.org/0000-0002-9961-7902)
- Jun Chang (ORCID: https://orcid.org/0000-0002-3594-9879)
Institutions
- Zhongnan University of Economics and Law (CN)
- Wuhan University (CN)
- Macquarie University (AU)
Publication Details
- Journal
- IEEE Transactions on Circuits and Systems for Video Technology
- Published
- 2024-08-05
- DOI
- https://doi.org/10.1109/tcsvt.2024.3438431
- Citations
- 2
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.70
Funders
- National Natural Science Foundation of China