RF-DET: Refocusing on the Small-scale Objects Using Aggregated Context for Accurate Power Transmitting Components Detection on UAV Oblique Imagery
This repo stores the code of the paper RF-DET: Refocusing on the Small-scale Objects Using Aggregated Context for Accurate Power Transmitting Components Detection on UAV Oblique Imagery.
In transmission lines, regular inspections are crucial for maintaining their safe operation. Automatic and accurate detection of power transmission facility components (power components) in inspection imagery is an effective way to monitor the status of electrical assets within the Right of Ways (RoWs). However, the multitude of small-scale objects (e.g. grading rings, vibration dampers) in inspection imagery poses enormous challenges. To address these challenges, we propose a coarse-to-fine object detector named RF-DET. It adopts a Refocus Framework to refine the detection accuracy of small objects within the Regions of Interest of the Power Components (P-RoIs) that are generated through explicit context. On the basis above, an Implicit Context Aggregation Attention Module (ICAM) is proposed. ICAM utilizes a multi-branch structure to capture and aggregate multi-directional positional and global information, enabling an in-depth mining of the implicit context among small objects. To verify the performance of this detector, a benchmark dataset named DOPI-UAV is constructed, which includes ample annotated instances, encompassing six common categories of power components and one category of defect. Experimental results show that RF-DET achieves mAP of 62.7%, 55.7%, 84.6%, and 52.8% on the DOPI-UAV, Tower, CPLID, and InsD datasets, respectively. Compared to the state-of-the-art method, such as YOLOv9, RF-DET attains significant performance improvements, with increases of 5.2% in mAP and 6.4% in mAP50, respectively. Especially, the APS is improved by 8.3%.
