Detection of Underground Dangerous Area Based on Improving YOLOV8

被引:6
作者
Ni, Yunfeng [1 ]
Huo, Jie [1 ]
Hou, Ying [1 ]
Wang, Jing [1 ]
Guo, Ping [1 ]
机构
[1] Xian Univ Sci & Technol, Coll Commun & Informat Engn, Xian 710054, Peoples R China
基金
中国国家自然科学基金;
关键词
dangerous area testing; YOLOV8; NMS; ray method; attention mechanism;
D O I
10.3390/electronics13030623
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In order to improve the safety needs of personnel in the dark environment under the well, this article adopts the improved YOLOV8 algorithm combined with the ray method to determine whether underground personnel are entering dangerous areas and to provide early warning. First of all, this article introduces the coordinate attention mechanism on the basis of YOLOV8 target detection so that the model pays attention to the location information of the target area so as to improve the detection accuracy of obstruction and small target areas. In addition, the Soft-Non-Maximum Suppression (SNMS) module is introduced to further improve accuracy. The improved model is then combined with the ray method to be deployed and applied under a variety of angles and different scenic information cameras. The experimental results show that the proposed method obtains 99.5% of the identification accuracy and a frame speed of 45 Frames Per Second (FPS) on the self-built dataset. Compared with the YOLOV8 model, it has a higher accuracy and can effectively cope with the changes and interference factors in the underground environment. Further, it meets the requirements for real-time testing in dangerous underground areas.
引用
收藏
页数:15
相关论文
共 39 条
  • [1] Cui L., 2023, Radio Eng, V53, P1375
  • [2] A comparative study of image low level feature extraction algorithms
    El-Gayar, M. M.
    Soliman, H.
    meky, N.
    [J]. EGYPTIAN INFORMATICS JOURNAL, 2013, 14 (02) : 175 - 181
  • [3] Fengbo W., 2023, P 3 INT S COMPUTER E
  • [4] Fu C.-Y., 2017, arXiv
  • [5] A federated calibration scheme for convolutional neural networks: Models, applications and challenges
    Gaba, Shivani
    Budhiraja, Ishan
    Kumar, Vimal
    Garg, Sahil
    Kaddoum, Georges
    Hassan, Mohammad Mehedi
    [J]. COMPUTER COMMUNICATIONS, 2022, 192 : 144 - 162
  • [6] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [7] Guo Chuanlei, 2018, Journal of Computer Applications, V38, P2833, DOI 10.11772/j.issn.1001-9081.2018030720
  • [8] He KM, 2020, IEEE T PATTERN ANAL, V42, P386, DOI [10.1109/TPAMI.2018.2844175, 10.1109/ICCV.2017.322]
  • [9] Hou Y., 2023, Coal, V32, P33
  • [10] Image classification and adversarial robustness analysis based on hybrid convolutional neural network
    Huang, Shui-Yuan
    An, Wan-Jia
    Zhang, De-Shun
    Zhou, Nan-Run
    [J]. OPTICS COMMUNICATIONS, 2023, 533