SHEN Ke, JI Liang, ZHANG Yuanhao, ZOU Sheng. Research on coal and gangue detection algorithm based on improved YOLOv5s model[J]. Journal of Mine Automation, 2021, 47(11): 107-111. DOI: 10.13272/j.issn.1671-251x.17838
Citation: SHEN Ke, JI Liang, ZHANG Yuanhao, ZOU Sheng. Research on coal and gangue detection algorithm based on improved YOLOv5s model[J]. Journal of Mine Automation, 2021, 47(11): 107-111. DOI: 10.13272/j.issn.1671-251x.17838

Research on coal and gangue detection algorithm based on improved YOLOv5s model

More Information
  • Received Date: August 30, 2021
  • Revised Date: November 15, 2021
  • In order to solve the problems of slow detection speed and low detection precision of the existing deep learning-based coal and gangue target detection methods, an improved YOLOv5s model is proposed and applied to coal and gangue target detection.The YOLOv5s model is improved by embedding self-calibrated convolutions(SCConv)in the Backbone area of YOLOv5s model as the characteristic extraction network, which can better fuse multi-scale characteristic information.Because the size of coal and gangue is too small compared with the whole image, the Neck area of YOLOv5s model is appropriately simplified, and the 19×19 characteristic map branches suitable for detecting larger size objects are deleted, thus reducing model complexity and improving the real-time detection performance.The anchor box obtained by clustering with K-means algorithm is linearly scaled to improve the model detection precision.The experiment of coal and gangue target detection based on improved YOLOv5s model shows that compared with YOLOv5s model, the improved YOLOv5s model can detect the corresponding coal and gangue accurately.The size of improved YOLOv5s model is reduced by 1.57 MB, the frame rate is increased by 2.1 frames/s, and the average precision is improved by 1.7%, indicating that the improved YOLOv5s model has improved both detection precision and detection speed.
  • [1]
    孙超,姜琳,袁广玉.“十四五”期间我国煤炭供需趋势分析[J].煤炭工程,2021,53(5):193-196.

    SUN Chao,JIANG Lin,YUAN Guangyu.Trend analysis of China's coal supply and demand during the 14th Five-Year Plan[J].Coal Engineering,2021,53(5):193-196.
    [2]
    ZHANG Ningbo,LIU Changyou.Radiation characteristics of natural gamma-ray from coal and gangue for recognition in top coal caving[J].Scientific Reports,2018,8(1):1-9.
    [3]
    MOHANTA K S,MEIKAPB C.Influence of medium particle size on the separation performance of an air dense medium fluidized bed separator for coal cleaning[J].Journal of the Southern African Institute of Mining and Metallurgy,2015,115(8):761-766.
    [4]
    李建平,郑克洪,杜长龙.煤和矸石的冲击破碎粒度分布特征[J].煤炭学报,2013,38(增刊1):54-58.

    LI Jianping,ZHENG Kehong,DU Changlong.The distribution discipline of impact crushed on coal and gangue[J].Journal of China Coal Society,2013,38(S1):54-58.
    [5]
    曹现刚,李莹,王鹏,等.煤矸石识别方法研究现状与展望[J].工矿自动化,2020,46(1):38-43.

    CAO Xiangang,LI Ying,WANG Peng,et al.Research status of coal-gangue identification method and its prospect[J].Industry and Mine Automation,2020,46(1):38-43.
    [6]
    刘富强,钱建生,王新红,等.基于图像处理与识别技术的煤矿矸石自动分选[J].煤炭学报,2000,25(5):534-537.

    LIU Fuqiang,QIAN Jiansheng,WANG Xinhong,et al.Automatic separation of waste rock in coal mine based on image procession and recognition[J].Journal of China Coal Society,2000,25(5):534-537.
    [7]
    刘丽,赵凌君,郭承玉,等.图像纹理分类方法研究进展和展望[J].自动化学报,2018,44(4):584-607.

    LIU Li,ZHAO Lingjun,GUO Chengyu,et al.Texture classification:state-of-the-art methods and prospects[J].Acta Automatica Sinica,2018,44(4):584-607.
    [8]
    ZHANG Ning,DONAHUE J,GIRSHICK R,et al.Part-based R-CNNs for fine-grained category detection[C]//European Conference on Computer Vision(ECCV),Zurich,2014:834-849.
    [9]
    LIU Wei,ANGUELOV D,ERHAN D,et al.SSD:single shot multibox detector[C]//European Conference on Computer Vision(ECCV),Amsterdam,2016:21-37.
    [10]
    REDMON J,DIVVALA S,GIRSHICK R,et al.You only look once:unified,real-time object detection[C]//Conference on Computer Vision and Pattern Recognition(CVPR),Las Vegas,2016:779-788.
    [11]
    王中举,毛馨凯,孙江.基于视频解析的智能煤矸分选技术研究[J].工矿自动化,2021,47(增刊1):122-125.

    WANG Zhongju,MAO Xinkai,SUN Jiang.Research on intelligent coal-gangue separation technology based on video analysis[J].Industry and Mine Automation,2021,47(S1):122-125.
    [12]
    来文豪,周孟然,胡锋,等.基于多光谱成像和改进YOLO v4的煤矸石检测[J].光学学报,2020,40(24):72-80.

    LAI Wenhao,ZHOU Mengran,HU Feng,et al.Coal gangue detection based on multi-spectral imaging and improved YOLO v4[J].Acta Optica Sinica,2020,40(24):72-80.
    [13]
    DU Shuangjiang,ZHANG Pin,ZHANG Baofu,et al.Weak and occluded vehicle detection in complex infrared environment based on improved YOLOv4[J].IEEE Access,2021,9:25671-25680.
    [14]
    JI Weizhen,LIU Deer,MENG Yifei,et al.Exploring the solutions via Retinex enhancements for fruit recognition impacts of outdoor sunlight:a case study of navel oranges[J].Evolutionary Intelligence,2021:1-37.
    [15]
    LIU Jiangjiang,HOU Qibin,CHENG Mingming,et al.Improving convolutional networks with self-calibrated convolutions[C]//Conference on Computer Vision and Pattern Recognition(CVPR),Seattle,2020:10093-10102.
  • Related Articles

    [1]CHEN Zhiwen, CHEN Ailiangfei, TANG Xiaodan, KE Haobin, JIANG Zhaohui, XIAO Fei. YOLOv5s pruning method for edge computing of coal mine safety monitoring[J]. Journal of Mine Automation, 2024, 50(7): 89-97. DOI: 10.13272/j.issn.1671-251x.2024010095
    [2]XU Jun, ZHAO Xiaohu, HOU Nianqi, WANG Jie, LIU Yulin. A maintenance guidance system for coal mine electromechanical equipment based on improved YOLOv5s[J]. Journal of Mine Automation, 2024, 50(5): 151-156. DOI: 10.13272/j.issn.1671-251x.2023090069
    [3]YAN Bijuan, WANG Kaimin, GUO Pengcheng, ZHENG Xinxu, DONG Hao, LIU Yong. Research on coal gangue detection in coal preparation plant based on YOLOv5s-FSW model[J]. Journal of Mine Automation, 2024, 50(5): 36-43, 66. DOI: 10.13272/j.issn.1671-251x.2023100090
    [4]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
    [5]HE Kai, CHENG Gang, WANG Xi, GE Qingnan, ZHANG Hui, ZHAO Dongyang. Research on coal gangue recognition method based on CED-YOLOv5s model[J]. Journal of Mine Automation, 2024, 50(2): 49-56, 82. DOI: 10.13272/j.issn.1671-251x.2023090065
    [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]ZHANG Hui, SU Guoyong, ZHAO Dongyang. Research on multi object detection in mining face based on FBEC-YOLOv5s[J]. Journal of Mine Automation, 2023, 49(11): 39-45. DOI: 10.13272/j.issn.1671-251x.2023060063
    [8]ZHANG Lei, WANG Haosheng, LEI Weiqiang, WANG Bin, LIN Jiangong. Coal gangue target detection of belt conveyor based on YOLOv5s-SDE[J]. Journal of Mine Automation, 2023, 49(4): 106-112. DOI: 10.13272/j.issn.1671-251x.2022080043
    [9]ZHANG Shiru, HUANG Zongliu, ZHANG Yuanhao, ZHANG Ao, JI Liang. Coal and gangue recognition research based on improved YOLOv5[J]. Journal of Mine Automation, 2022, 48(11): 39-44. DOI: 10.13272/j.issn.1671-251x.2022060052
    [10]ZHANG Xuhui, YAN Jianxing, ZHANG Chao, WAN Jicheng, WANG Lixin, HU Chengjun, WANG Li, WANG Dong. Coal block abnormal behavior identification based on improved YOLOv5s + DeepSORT[J]. Journal of Mine Automation, 2022, 48(6): 77-86, 117. DOI: 10.13272/j.issn.1671-251x.17915

Catalog

    Article Metrics

    Article views (277) PDF downloads (64) Cited by()
    Related

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return