一种基于改进YOLOv5模型的炮孔识别方法
摘要:
本发明涉及计算机视觉领域的目标检测任务,尤其涉及一种基于改进YOLOv5模型的炮孔识别方法,通过基于改进YOLOv5的目标检测模型,制作数据集;标注出炮孔图片中工作面上的全部炮孔;提取出炮孔图片中每一个炮孔的中心点的坐标;根据提取到的炮控中心的坐标进行绘制实际炮孔布局图。改进的YOLOv5模型包括添加注意力机制,更多关注炮孔图片上炮孔位置的特征;改进回归损失函数,使得模型预测框的位置更加接近于真实框的位置。通过该方法,增强了最终训练模型的鲁棒性,提高了炮孔识别模型的准确率,可以快速实现炮孔的标定和位置坐标的获取,为井巷掘进爆破的工作减少人力和物力,提高工作人员的工作效率。
0/0