YE Ou, DOU Xiaoyi, FU Yan, DENG Jun. Coal block detection method integrating lightweight network and dual attention mechanism[J]. Journal of Mine Automation, 2021, 47(12): 75-80. DOI: 10.13272/j.issn.1671-251x.2021030075
Citation: YE Ou, DOU Xiaoyi, FU Yan, DENG Jun. Coal block detection method integrating lightweight network and dual attention mechanism[J]. Journal of Mine Automation, 2021, 47(12): 75-80. DOI: 10.13272/j.issn.1671-251x.2021030075

Coal block detection method integrating lightweight network and dual attention mechanism

More Information
  • Received Date: March 22, 2021
  • Revised Date: December 18, 2021
  • In order to solve the problems of low detection precision and slow detection speed of existing coal block detection methods on belt conveyor in underground coal mine, an improved YOLOv4 model integrating lightweight network and dual attention mechanism is proposed, and it is applied to coal block detection of belt conveyor. The improved YOLOv4 model uses K-means clustering algorithm to re-cluster the prior frames, so that the prior frames are more suitable for the size of the detected target. The model improves the backbone network structure by introducing the MobileNet lightweight network model to reduce the amount of model parameters and calculations, and improve the detection speed. A convolution block attention module with dual attention mechanism is embedded to improve the sensitivity of the model to target characteristics, suppress interference information and improve the precision of target detection. The experimental results show that the improved YOLOv4 model can detect coal blocks of different sizes accurately. Compared with the YOLOv4 model, the improved YOLOv4 model weight file is reduced by 36.46%, the accuracy rate is increased by 2.16%, the recall rate is increased by 20.4%, the average accuracy is increased by 14.37%, the missed detection rate is decreased by 16%, the detection speed is increased by 19 frames/s, the processing time for a single image is reduced by 1.31 s, which improves the detection precision and speed of coal block detection.
  • [1]
    李占利,陈佳迎,李洪安,等.胶带输送机智能视频检测与预警方法[J].图学学报,2017,38(2):230-235.

    LI Zhanli,CHEN Jiaying,LI Hong'an,et al.Research on intelligent monitoring and warning method of belt conveyor[J].Journal of Graphics,2017,38(2):230-235.
    [2]
    徐青云,赵耀江,李永明.我国煤矿事故统计分析及今后预防措施[J].煤炭工程,2015,47(3):80-82.

    XU Qingyun,ZHAO Yaojiang,LI Yongming.Statistical analysis and precautions of coal mine accidents in China[J].Coal Engineering,2015,47(3):80-82.
    [3]
    HU Chuan,CAO Huiping.Aspect-level influence discovery from graphs[J].IEEE Transactions on Knowledge & Data Engineering,2016,28(7):1635-1649.
    [4]
    WU Jianxin,YANG Hao.Linear regression-based efficient SVM learning for large-scale classification[J]. IEEE Transactions on Neural Networks & Learning Systems,2015,26(10):2357-2369.
    [5]
    FORSYTH D.Object detection with discriminatively trained part-based models[J].Computer,2014,47(2):6-7.
    [6]
    贾建英,董安国.基于联合直方图的运动目标检测算法[J].计算机工程与应用, 2016,52(5):199-203.

    JIA Jianying,DONG Anguo.Moving target detection algorithm based on joint histogram[J].Computer Engineering and Applications,2016,52(5):199-203.
    [7]
    LE M,WOO B,JO K.A comparison of SIFT and Harris conner features for correspondence points matching[C]//The 17th Korea-Japan Joint Workshop on Frontiers of Computer Vision(FCV),Ulsan,2017:1-4.
    [8]
    吕志强.复杂环境下煤矿皮带运输异物图像识别研究[D].徐州:中国矿业大学,2020.

    LYU Zhiqiang. Research on image recognition of foreign bodies in the process of coal mine belt transportation in complex environment[D].Xuzhou:China University of Mining and Technology,2020.
    [9]
    WANG Yuanbin,WANG Yujing,DANG Langfei.Video detection of foreign objects on the surface of belt conveyor underground coal mine based on improved SSD[J].Journal of Ambient Intelligence and Humanized Computing,2020(9):1-10.
    [10]
    胡璟皓,高妍,张红娟,等.基于深度学习的带式输送机非煤异物识别方法[J].工矿自动化, 2021,47(6):57-62.

    HU Jinghao,GAO Yan,ZHANG Hongjuan,et al. Research on the identification method of non-coal foreign object of belt conveyor based on deep learning[J]. Industry and Mine Automation,2021,47(6):57-62.
    [11]
    杜京义,陈瑞,郝乐,等.煤矿带式输送机异物检测[J].工矿自动化,2021,47(8):77-83.

    DU Jingyi,CHEN Rui,HAO Le,et al.Coal mine belt conveyor foreign object detection[J].Industry and Mine Automation,2021,47(8):77-83.
    [12]
    张伟,庄幸涛,王雪力,等.DS-YOLO:一种部署在无人机终端上的小目标实时检测算法[J].南京邮电大学学报(自然科学版),2021,41(1):86-98.

    ZHANG Wei,ZHUANG Xingtao,WANG Xueli,et al. DS-YOLO:a real-time small object detection algorithm on UAVs[J].Journal of Nanjing University of Posts and Telecommunications(Natural Science Edition),2021,41(1):86-98.
    [13]
    MITTAL S. A survey on optimized implementation of deep learning models on the NVIDIA Jetson platform[J]. Journal of Systems Architecture, 2019, 97:428-442.
    [14]
    HOWARD A G,ZHU Menglong,CHEN Bo,et al.MobileNets efficient convolutional neural networks for mobile vision applications[EB/OL].(2018-01-22)[2021-03-23]. https://arxiv.org/abs/1704.04861 arXiv:1704.04861.2017.
    [15]
    WOO S,PARK J,LEE J Y,et al.CBAM:convolutional block attention module[C]//2018 ECCV Conference on Computer Vision,Berlin,2018:3-19.
    [16]
    KANUNGO T,MOUNT D,NETANYAHU N,et al.A local search approximation algorithm for k-means clustering[J].Computational Geometry,2004,28(23):89-112.
    [17]
    WANG C Y,LIAO H Y M,WU Y H,et al.CSPNet:a new backbone that can enhance learning capability of CNN[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops(CVPRW),Seattle,2020:1571-1580.
  • Related Articles

    [1]Foreign Object Detection Method for Underground Mine Conveyor Belts Based on Extremely Lightweight YOLOv8n[J]. Journal of Mine Automation.
    [2]XU Ciqiang, JIA Yunhong, TIAN Yuan. Large block coal detection algorithm for fully mechanized working face based on MES-YOLOv5s[J]. Journal of Mine Automation, 2024, 50(3): 42-47, 141. DOI: 10.13272/j.issn.1671-251x.2024030009
    [3]QIN Yulong, CHENG Jiming, REN Yige, WANG Xiaoqing, ZHAO Qing, AN Cuijuan. Large coal detection for belt conveyors based on improved YOLOv5[J]. Journal of Mine Automation, 2024, 50(2): 57-62, 71. DOI: 10.13272/j.issn.1671-251x.2023080096
    [4]CAO Zhengyuan, JIANG Wei, FANG Chenghui. Intelligent detection method for coal flow foreign objects based on dual attention generative adversarial network[J]. Journal of Mine Automation, 2023, 49(12): 56-62. DOI: 10.13272/j.issn.1671-251x.18094
    [5]LI Zhongfei, FENG Shiyong, GUO Jun, ZHANG Yunhe, XU Feixiang. Lightweight safety helmet wearing detection fusing coordinate attention and multiscale feature[J]. Journal of Mine Automation, 2023, 49(11): 151-159. DOI: 10.13272/j.issn.1671-251x.2023080123
    [6]ZHAO Wei, WANG Shuang, ZHAO Dongyang. Multi object detection of underground unmanned electric locomotives in coal mines based on SD-YOLOv5s-4L[J]. Journal of Mine Automation, 2023, 49(11): 121-128. DOI: 10.13272/j.issn.1671-251x.2023070100
    [7]ZHU Fuwen, HOU Zhihui, LI Mingzhen. Lightweight multi-scale cross channel attention coal flow detection network[J]. Journal of Mine Automation, 2023, 49(8): 100-105. DOI: 10.13272/j.issn.1671-251x.2023030045
    [8]WANG Keping, LIAN Kaihai, YANG Yi, FEI Shumin. Target detection of the fully mechanized working face based on improved YOLOv4[J]. Journal of Mine Automation, 2023, 49(2): 70-76. DOI: 10.13272/j.issn.1671-251x.2022070080
    [9]JIN Shukai, WEI Guannan, WANG Chunming, WANG Tonghai, WU Zhonglun, YANG Kehu. Intelligent identification method for mine car load in coal mine auxiliary shaft[J]. Journal of Mine Automation, 2022, 48(4): 14-19, 30. DOI: 10.13272/j.issn.1671-251x.2021110055
    [10]ZHOU Yujie, XU Shanyong, HUANG Yourui, TANG Chaoli. Conveyor belt damage detection method based on improved YOLOv4[J]. Journal of Mine Automation, 2021, 47(11): 61-65. DOI: 10.13272/j.issn.1671-251x.17843

Catalog

    DENG Jun

    1. On this Site
    2. On Google Scholar
    3. On PubMed

    Article Metrics

    Article views (214) PDF downloads (34) Cited by()
    Related

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return