[发明专利]一种烟叶布料车和传送带的位置确定方法及装置

说明书

本申请提供了一种烟叶布料车和传送带的位置确定方法及装置，其中，该方法包括：获取包含烟叶布料车和传送带的图像；确定图像中的烟叶布料车和传送带分别对应的包围框信息和修正包围框信息；将修正包围框信息与对应的包围框信息的差值，确定为包围框信息的补偿；将图像作为训练样本、将图像对应的包围框信息和补偿作为标签，输入到神经网络模型中进行训练；将包含待确定烟叶布料车和待确定传送带的待确定图像输入至训练好的神经网络模型中，预测出待确定图像的预测包围框信息和预测补偿；根据预测包围框信息和预测补偿，确定待确定烟叶布料车和待确定传送带的位置。

技术领域

本申请涉及计算机技术领域，尤其涉及一种烟叶布料车和传送带的位置确定方法及装置。

背景技术

在烟草生成工厂中有很多烟叶布料车和传送带，为了保证传输过程的安全性，需要使烟叶布料车和传送带的位置满足一定关系。现有技术中，工作人员通过摄像头观察烟叶布料车和传送带的位置关系，进而可能产生判断烟叶布料车和传送带的位置关系错误的情况，影响传输安全和效率。

发明内容

有鉴于此，本申请的目的在于至少提供一种烟叶布料车和传送带的位置确定方法及装置，通过烟叶布料车和传送带的标记位置及修正位置得到的补偿来训练神经网络模型，使得神经网络模型可以更加准确地识别出烟叶布料车和传送带的位置，解决了现有技术中烟叶布料车和传送带的位置识别准确率低的技术问题，达到了提高烟叶布料车和传送带的位置识别准确率的技术效果。

本申请主要包括以下几个方面：

第一方面，本申请实施例提供一种烟叶布料车和传送带的位置确定方法，该方法包括：获取包含烟叶布料车和传送带的图像；确定图像中的烟叶布料车和传送带分别对应的包围框信息和修正包围框信息；包围框信息是通过标注软件确定的烟叶布料车和传送带分别对应的包围框的大小和位置；修正包围框信息为人为标注的烟叶布料车和传送带分别对应的包围框的大小和位置；将修正包围框信息与对应的包围框信息的差值，确定为包围框信息的补偿；将图像作为训练样本、将图像对应的包围框信息和补偿作为标签，输入到神经网络模型中进行训练；将包含待确定烟叶布料车和待确定传送带的待确定图像输入至训练好的神经网络模型中，预测出待确定图像的预测包围框信息和预测补偿；根据预测包围框信息和预测补偿，确定待确定烟叶布料车和待确定传送带的位置。

可选地，预测包围框信息包括：预测包围框的预测中心点坐标、预测中心点偏移、预测宽度、预测高度；预测补偿包括：预测中心点坐标补偿、预测中心点偏移补偿、预测宽度补偿、预测高度补偿；根据预测包围框信息和预测补偿，确定待确定烟叶布料车和待确定传送带的位置包括：将待确定烟叶布料车和待确定传送带分别对应的预测中心点坐标、预测中心点偏移、预测宽度、预测高度与预测中心点坐标补偿、预测中心点偏移补偿、预测宽度补偿、预测高度补偿对应相加，确定为待确定烟叶布料车和待确定传送带分别对应的位置。

可选地，根据预测包围框信息和预测补偿，确定待确定烟叶布料车和待确定传送带的位置之后，方法还包括：计算待确定烟叶布料车和待确定传送带的重合的高度值；判断重合的高度值与待确定烟叶布料车对应的预测高度的比值是否大于预设值；若重合的高度值与待确定烟叶布料车对应的预测高度的比值大于预设值，则确定待确定烟叶布料车和待确定传送带的位置正常。

可选地，神经网络模型的损失函数通过以下公式计算：

L_det＝L_c+L_k+λ_sizeL_size+L_off+L_class