基于跨阶段部分网络改进目标检测算法的煤矸识别方法
摘要:
本发明公开了基于跨阶段部分网络改进目标检测算法的煤矸识别方法,步骤包括:井下工作面采集煤矸图像进行预处理得到煤矸数据集;通过数据增强方法扩充煤矸数据集,并划分数据集得到训练集和测试集;采用跨阶段部分网络改进YOLOv7目标检测算法构建煤矸识别模型;采用三次迁移学习对模型进行训练;使用训练好的模型对煤矸测试集进行分类识别与位置检测。本方法通过改进YOLOv7目标检测算法实现了一种新型的煤矸识别模型,该煤矸识别模型的参数量更小同时对煤矸识别的精度更高,在模型的识别精度和速度之间取得了较好的平衡。
0/0