Heuristic data-driven anchor generation for UAV-based maritime rescue image object detection

Abstract

The specificity of scenarios and tasks in Unmanned Aerial Vehicles (UAV)-based maritime rescue poses challenges for detecting targets within images captured by drones in such environments. This study focuses on leveraging heuristic methods to extract data features from specific UAV maritime rescue images to optimize the generation of anchor boxes in detection models. Experiments conducted on the large-scale SeaDronesSee maritime rescue dataset, using the MMDetection object detection framework, demonstrated that the optimized anchor boxes, improved model performance by 48.9% to 62.8% compared to the framework's default configuration, with the most proficient model surpassing the official highest SeaDronesSee baseline by over 49.3%. Further analysis of the results revealed the variation in detection difficulty for different objects within the dataset and identified the reasons behind these differences. The methodology and analysis presented in this study hold promise for optimizing UAV-based maritime rescue object detection models as well as refining data analysis and enhancement.