Real-time detection of rockfall on slopes is an essential part of a smart worksite. As a result, target detection techniques for rockfall detection have been rapidly developed. However, the complex geologic environment of slopes, special climatic conditions, and human factors pose significant challenges to this research. In this paper, we propose an enhanced high-speed slope rockfall detection method based on YOLOv8n. First, the LSKAttention mechanism is added to the backbone part to improve the model’s ability to balance the processing of global and local information, which enhances the model’s accuracy and generalization ability. Second, in order to ensuredetection accuracy for smaller targets, an enhanced detection head is added, and other detection heads of different sizes are combined to form a multi-scale feature fusion to improve the overall detection performance. Finally, a bidirectional feature pyramid network (BiFPN) is introduced in the neck to effectively reduce the parameters and computational complexity and improve the overall performance of rockfall detection. In addition we compare the LSKAttention mechanism with other attention mechanisms to verify the effectiveness of the improvements. Compared with the baseline model, our method improves the average accuracy mAP@0.5 by 4.8%. Moreover, the amount of parameters is reduced by 20.2%. Among the different evaluation criteria, the LHB-YOLOv8 method shows obvious advantages, making it suitable for engineering applications and the practical deployment of slope rockfall detection systems.
Read full abstract