[发明专利]一种用于工业检测的图像分类方法、装置及计算机设备

摘要

本申请涉及深度学习技术领域，具体而言，涉及一种用于工业检测的图像分类方法、装置及计算机设备，一定程度上可以解决由于图像差异性小，对图像进行更精细的子类划分时存在分类检测准确度低的问题。所述用于工业检测的图像分类方法包括：获取第一训练图像的特征向量，并将所述特征向量输入至具有多个类别向量的权重矩阵，类别向量根据图像分类需求设定；扩大角度空间中各初始向量夹角，得到目标向量夹角对应的余弦值，所述初始向量夹角为所述特征向量与各所述类别向量之间的角度；其中，所述余弦值用于确定相似度，所述角度空间是基于所述权重矩阵转化得到的分类空间；根据所述相似度所在的对应阈值区间，确定所述第一训练图像对应的第一类别。