Aiming at the complex environment such as uneven illumination and variable target scales in mines, achieving precise and realtime detection of potential collision targets such as personnel and equipment by autonomous mining vehicles is crucial for improving mine safety and work efficiency. To address the above problems, this paper proposes an improved algorithm based on YOLOv8. The Robust Feature Downsampling (RFD) module is introduced to solve the problem of small target feature loss, and the weighted Bidirectional Feature Pyramid Network (BiFPN) is adopted to enhance the multi-scale feature fusion capability. Experimental results show that the improved model achieves detection accuracies of 91.5% mAP and 86.1% F1-score on the self-built mine scene dataset, which are 3.3% and 2.6% higher than the original benchmark model YOLOv8-n, respectively. Meanwhile, the model complexity is significantly reduced. This algorithm ensures the efficient and accurate operation of the real-time obstacle detection system for unmanned loaders.
Bochkovskiy, A., Wang, C.Y. and Liao, H.Y.M. (2020) YOLOv4: Optimal Speed and Accuracy of Object Detection. arXiv: 2004.10934.
[3]
Redmon, J., Divvala, S., Girshick, R. and Farhadi, A. (2016) You Only Look Once: Unified, Real-Time Object Detection. 2016 IEEE Con-ference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, 27-30 June 2016, 779-788. https://doi.org/10.1109/cvpr.2016.91
[4]
Redmon, J. (2018) YOLOv3: An Incremental Improvement. arXiv: 1804.02767.
[5]
Redmon, J. and Farhadi, A. (2017) YOLO9000: Better, Faster, Stronger. 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, 21-26 July 2017, 6517-6525. https://doi.org/10.1109/cvpr.2017.690
Lu, W., Chen, S., Tang, J., Ding, C.H.Q. and Luo, B. (2023) A Robust Feature Downsampling Module for Remote-Sensing Visual Tasks. IEEE Transactions on Geoscience and Remote Sensing, 61, 1-12. https://doi.org/10.1109/tgrs.2023.3282048
[12]
Tan, M., Pang, R. and Le, Q.V. (2020) EfficientDet: Scalable and Efficient Object Detection. 2020 IEEE/CVF Conference on Computer Vision and Pat-tern Recognition (CVPR), Seattle, 13-19 June 2020, 10778-10787. https://doi.org/10.1109/cvpr42600.2020.01079