An Objectness-Aware Network for Wildlife Detection

Xin Lu,Weixuan Li,Xiaobo Lu
DOI: https://doi.org/10.1007/s11042-023-15246-8
IF: 2.577
2024-01-01
Multimedia Tools and Applications
Abstract:The application of object detectors contributes to the monitoring and protection of wildlife with high accuracy and fast speed. However, the variety of shapes and appearances of wildlife bring difficulties to the detection task, which requires the detection models a better objectness definition and label assignment. This study proposes an objectness-aware YOLO (OA-YOLO) to improve the effect on wildlife detection. First, we redefine the objectness values of training samples and decouple the objectness branch in the detector head. Second, we propose a Natural Breaks Label Assignment (NBLA) algorithm to divide the anchors into positive, negative, and assistant samples automatically based on their objectness values. Third, the assistant samples used to be ignored participate in the classification training in an objectness-related weighted manner to improve the detection accuracy. The experimental results indicate that OA-YOLO improves the mean average precision (mAP) by 6.9% on the challenging wildlife dataset and outperforms the existing approaches.
What problem does this paper attempt to address?