[发明专利]模型输入尺寸确定方法及相关设备

说明书

本申请涉及图像检测，提供一种模型输入尺寸确定方法及相关设备。该方法能够对测试图像进行编码，得到编码向量并进行解码，得到重构图像，比较重构图像与测试图像，得到重构误差，按照多个预设尺寸切分编码向量，得到每个预设尺寸的多个子向量，将多个子向量输入至高斯混合模型中，得到每个预设尺寸的多个子概率，确定每个预设尺寸的估测概率及测试误差，根据每个测试误差及每个误差阈值确定检测结果及准确度，根据准确度确定输入尺寸。本申请通过确定出适合瑕疵检测的模型输入尺寸，从而提高瑕疵检测的准确度。

技术领域

本申请涉及图像检测技术领域，尤其涉及一种模型输入尺寸确定方法及相关设备。

背景技术

为了提高工业产品的品质，在对工业产品进行打包前，通常会对工业产品进行一定的瑕疵检测。在对产品进行整体检测时，当产品中的瑕疵较为细微，会导致难以检测到，从而降低对图像进行瑕疵检测的准确度。

发明内容

鉴于以上内容，有必要提供一种模型输入尺寸确定方法及相关设备，能够检测出细微瑕疵的存在，从而提高瑕疵检测的准确度。

本申请的第一方面提供一种模型输入尺寸确定方法，所述模型输入尺寸确定方法包括：

获取测试图像集，所述测试图像集包括测试图像及瑕疵结果；

对所述测试图像进行编码处理，得到编码向量；

对所述编码向量进行解码处理，得到所述测试图像的重构图像，并比较所述重构图像与所述测试图像，得到所述测试图像的重构误差；

按照多个预设尺寸切分所述编码向量，得到每个预设尺寸对应的多个子向量；

将每个预设尺寸对应的多个子向量输入至预先训练好的高斯混合模型中，得到每个预设尺寸的多个子概率，并根据每个预设尺寸的多个子概率确定每个预设尺寸的估测概率；

根据每个预设尺寸的估测概率及所述重构误差确定每个预设尺寸的测试误差，并根据每个预设尺寸的测试误差及每个预设尺寸的误差阈值确定每个预设尺寸下的测试图像的检测结果；

根据每个预设尺寸的检测结果及所述瑕疵结果确定每个预设尺寸的准确度，并根据每个预设尺寸的准确度从所述多个预设尺寸中选取输入尺寸。

根据本申请可选实施例，所述对所述编码向量进行解码处理，得到所述测试图像的重构图像包括：

获取向量映射表；

根据所述向量映射表将所述编码向量映射为多个重构像素点；

组合所述多个重构像素点，得到所述重构图像。

根据本申请可选实施例，所述比较所述重构图像与所述测试图像，得到所述测试图像的重构误差包括：

从所述测试图像中提取与每个重构像素点对应的测试像素点；

比较每个重构像素点与对应的测试像素点，并确定与对应的测试像素点不同的重构像素点的数量作为第一数量；

计算所述多个重构像素点的数量，得到第二数量；

将所述第一数量除以所述第二数量，得到所述重构误差。

根据本申请可选实施例，所述按照多个预设尺寸切分所述编码向量，得到每个预设尺寸对应的多个子向量包括：

按照每个预设尺寸对所述编码向量进行顺序切分，得到每个预设尺寸对应的多个子向量。

根据本申请可选实施例，所述将每个预设尺寸对应的多个子向量输入至预先训练好的高斯混合模型中，得到每个预设尺寸的多个子概率，并根据每个预设尺寸的多个子概率确定每个预设尺寸的估测概率包括：

获取所述高斯混合模型中的特征分布及混合系数；