[发明专利]一种物料信息的确定方法、装置、设备及存储介质

说明书

本发明实施例提供一种物料信息的确定方法、装置、设备及存储介质，涉及图像处理领域，能够提高确定物料信息的处理速度。该方法包括：获取待处理灰度图像；待处理灰度图像包括对包括物料盘以及物料的待处理射线图像执行预处理操作得到的，预处理操作包括位图转换和/或灰度拉伸；按照预设尺寸，对待处理灰度图像进行裁剪处理，以生成多个待处理子灰度图像；待处理灰度图像由多个待处理子灰度图像组成；基于多个待处理子灰度图像以及预设的图像处理模型，生成与多个待处理子灰度图像一一对应的多个目标子二值图像；根据生成到的多个目标子二值图像，确定待处理射线图像中物料的数量及物料在物料盘上的位置。

技术领域

本发明涉及图像处理领域，尤其涉及一种物料信息的确定方法、装置、设备及存储介质。

背景技术

现有技术中，可以利用卷积神经网络统计表面贴装技术(Surface MountedTechnology，SMT)领域中使用的物料(又称为贴片元器件，包括贴片电感、贴片电容、贴片电阻等)的物料信息。具体的，用于确定物料信息的设备获取包括物料以及物料盘的X射线图像，并将X射线图像进行处理，以得到一个灰度图像，并将得到的灰度图像输入到预先训练好的卷积神经网络中，以获取卷积神经网络输出的二值图像，进而可以从获取到的二值图像中确定物料的数量以及位置。

在上述物料信息的确定过程中，由于对X射线图像处理得到的灰度图像分辨率非常大，这就导致对设备的图形处理器(Graphics Processing Unit，GPU)和显存的硬件资源要求较高，处理速度慢。

发明内容

本发明的实施例提供一种物料信息的确定方法、装置、设备及存储介质，能够提高确定物料信息的处理速度。

为达到上述目的，本发明的实施例采用如下技术方案：

第一方面，提供一种物料信息的确定方法，包括：获取待处理灰度图像；待处理灰度图像包括对包括物料盘以及物料的待处理射线图像执行预处理操作得到的，预处理操作包括位图转换和/或灰度拉伸；按照预设尺寸，对待处理灰度图像进行裁剪处理，以生成多个待处理子灰度图像；待处理灰度图像由多个待处理子灰度图像组成；基于多个待处理子灰度图像以及预设的图像处理模型，生成与多个待处理子灰度图像一一对应的多个目标子二值图像；根据生成到的多个目标子二值图像，确定待处理射线图像中物料的数量及物料在物料盘上的位置。

基于上述方案，能够对待处理灰度图像进行裁剪处理，并基于裁剪得到的多个待处理子灰度图像以及图像处理模型，生成与多个待处理子灰度图像一一对应的多个目标子二值图像，并根据生成的多个目标子二值图像，确定物料的数量以及物料的位置。由于本方案能够采用图像处理模型对裁剪得到的多个待处理子灰度图像进行异步的处理，因此，相对于现有的对大分辨率的灰度图像同步处理的过程，能够提高处理速度。

进一步可选的，上述根据生成到的多个目标子二值图像，确定待处理射线图像中物料的数量及物料在物料盘上的位置，包括：合并多个目标子二值图像，以得到目标二值图像；目标二值图像的尺寸与待处理灰度图像的尺寸相同；确定目标二值图像中的连通区域，以确定物料的数量以及物料在物料盘上的位置；一个连通区域用于表征一个物料。

进一步可选的，上述物料的确定方法，还包括：获取多个样本子灰度图像以及多个样本子二值图像；多个样本子二值图像与多个样本子灰度图像一一对应；多个样本子灰度图像为裁剪包括物料盘以及物料的样本灰度图像后，得到的用于反映物料盘以及物料的特征的灰度图像；基于多个样本子灰度图像以及多个样本子二值图像，训练得到图像处理模型；图像处理模型用于确定与输入的待处理子灰度图像对应的目标子二值图像。