(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210857094.9 (22)申请日 2022.07.20 (71)申请人 凌云光技术股份有限公司 地址 100094 北京市海淀区翠湖南环路13 号院7号楼7层701室 (72)发明人 张黎　 (74)专利代理 机构 北京弘权知识产权代理有限 公司 11363 专利代理师 逯长明　许伟群 (51)Int.Cl. G06T 7/00(2017.01) G06T 7/62(2017.01) G06V 10/44(2022.01) G06V 10/764(2022.01) G06N 3/08(2006.01)G06N 3/04(2006.01) (54)发明名称 一种用于工业检测的图像 分类方法、 装置及 计算机设备 (57)摘要 本申请涉及深度学习技术领域， 具体而言， 涉及一种用于工业检测的图像分类方法、 装置及 计算机设备， 一定程度上可以解决由于图像差异 性小， 对图像进行更精细的子类划分时存在分类 检测准确度低的问题。 所述用于工业检测的图像 分类方法包括： 获取第一训练图像的特征向量， 并将所述特征向量输入至具有多个类别向量的 权重矩阵， 类别向量根据图像分类需求设定； 扩 大角度空间中各初始向量夹角， 得到目标向量夹 角对应的余弦值， 所述初始向量夹角为所述特征 向量与各所述类别向量之间的角度； 其中， 所述 余弦值用于确定相似度， 所述角度空间是基于所 述权重矩阵转化得到的分类空间； 根据所述相似 度所在的对应阈值区间， 确定所述第一训练图像 对应的第一类别。 权利要求书2页 说明书8页 附图2页 CN 114972335 A 2022.08.30 CN 114972335 A 1.一种用于 工业检测的图像分类方法， 其特 征在于， 所述方法包括： 获取第一训练图像的特征向量， 并将所述特征向量输入至具有多个类别向量的权重矩 阵， 所述类别向量 根据图像分类需求设定； 扩大角度空间中各初始向量夹角， 得到目标向量夹角对应的余弦值， 所述初始向量夹 角为所述特征向量与各所述类别向量之 间的角度； 其中， 所述余弦值用于确定相似度， 所述 角度空间是基于所述权 重矩阵转 化得到的分类空间； 根据所述相似度所在的对应阈值区间， 确定所述第一训练图像对应的第一类别。 2.如权利要求1所述的用于工业检测的图像分类方法， 其特征在于， 所述扩大角度空间 中各初始向量夹角， 得到目标向量夹角对应的余弦值， 进一 步包括： 设定分离角度间隔； 求取分离角度间隔与 所述初始向量夹角之和、 或求取分离角度间隔与 所述初始向量夹 角之积， 得到目标向量夹角； 计算所述目标向量夹角的余弦值。 3.如权利要求2所述的用于工业检测的图像分类方法， 其特征在于， 计算所述初始向量 夹角， 进一 步包括： 计算所述权 重矩阵中的每一个所述类别向量与所述特 征向量的点积； 基于所述点积， 计算所述特征向量与每一个所述类别向量之间的所述初始向量夹角， 所述初始向量夹角是 大小为n的一维向量， 所述 n表示类别向量的类别数。 4.如权利要求3所述的用于工业检测的图像分类方法， 其特征在于， 所述确定所述相似 度， 进一步包括： 通过交叉熵损失函数计算所述相似度， 所述相似度用于表征所述余弦值与 所述类别向 量之间的相似程度。 5.如权利要求4所述的用于工业检测的图像分类方法， 其特征在于， 所述角度空间是基 于所述权 重矩阵转 化得到的分类空间， 进一 步包括： 计算特征向量模和所述权 重矩阵的行参数模； 计算所述特征向量与 所述特征向量模的比值、 以及所述权重矩阵的行参数与 所述行参 数模的比值， 求取两个比值之积为角度余弦值； 通过所述角度余弦值反余弦计算得到角度， 实现将所述权重矩阵转化到所述角度空 间。 6.如权利要求1所述的用于工业检测的图像分类方法， 其特征在于， 获取第 一训练图像 的特征向量， 进一 步包括： 通过主干模型从所述第 一训练图像中提取所述特征向量， 所述特征向量是大小为所述 n的一维向量。 7.如权利要求1所述的用于工业检测的图像分类方法， 其特征在于， 在所述将所述特征 向量输入至具有 多个类别向量的权 重矩阵之后， 所述方法还 包括： 将所述类别向量按照当前索引 位所针对的类别划分为当前索引位类别和其他索引位 类别， 其中， 将所述当前索引位类别表示 为1， 将所述 其他索引位类别表示 为0。 8.一种用于 工业检测的图像分类装置， 其特 征在于， 包括： 获取模块， 用于获取第一训练图像的特征向量， 并将所述特征向量输入至具有多个类权　利　要　求　书 1/2 页 2 CN 114972335 A 2别向量的权 重矩阵， 所述类别向量 根据图像分类需求设定； 扩大模块， 用于扩大角度空间中各初始向量夹角， 得到目标向量夹角对应的余弦值， 所 述初始向量夹角为所述特征向量与各所述类别向量之间的角度； 其中， 所述余弦值用于确 定相似度， 所述角度空间是基于所述权 重矩阵转 化得到的分类空间； 确定模块， 用于根据所述相似度所在的对应阈值区间， 确定所述第一训练图像对应的 第一类别。 9.如权利要求8所述的用于工业检测的图像分类装置， 其特征在于， 所述扩大模块包 括： 设定单元， 用于设定分离角度间隔； 第一计算单元， 用于求取分离角度间隔与所述初始向量夹角之和、 或求取分离角度间 隔与所述初始向量夹角之积， 得到目标向量夹角； 第二计算单 元， 用于计算所述目标向量夹角的余弦值。 10.一种计算机设备， 包括存储器和 处理器， 所述存储器存储有计算机程序， 其特征在 于， 所述处 理器执行所述计算机程序时实现权利要求1至7中任一项所述的方法的步骤。权　利　要　求　书 2/2 页 3 CN 114972335 A 3