煤炭工程 ›› 2023, Vol. 55 ›› Issue (12): 161-166.doi: 10.11799/ce202312027
崔 斌,陈 林,亓玉浩,张 坤,赵得福,黄梁松,李明霞,孔祥俊,杜明超,蒋祥卿,刘 源
1. 山东科技大学 山东省机器人与智能技术重点实验室,山东 青岛 266590
2. 欧普瑞泰环境科技有限公司,山东 青岛 266200
3. 青岛华夏橡胶工业有限公司,山东 青岛 266200
4. 北斗天地股份有限公司,山东 济宁 710000
5. 青海能源发展(集团)有限公司,青海 西宁 810008
摘要:
为提高煤矸石分拣的精度和可靠性,提出了一种基于改进YOLO v4的煤矸石识别网络,引入了Focal损失函数,使用K-means++聚类算法优化初始锚定框,将PANet中的五次卷积操作替换为CSP结构,同时引入空洞卷积的金字塔结构,降低模型参数, 实现模型的轻量化,增加了一条跨连接边构成BiFPN结构,提高对中等目标的检测能力,得到My-YOLO v4目标检测模型。本研究对所提出的My-YOLO v4识别检测方法与SSD、YOLO v3、YOLO v4三种检测方法进行实验对比分析。实验结果表明,该检测算法在测试集上检测煤与煤矸石混合的mAP值为98.14%,FPS为28.3f/s,相较于SSD、YOLO v3检测算法识别精度分别提高了5.41%、2.87%,相较于YOLO v4目标检测模型识别速度提高了7.7f/s,通过对比分析实验数据验证了My-YOLO v4目标检测模型整体性能的有效提高。
中图分类号: