(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202211298870.2 (22)申请日 2022.10.24 (71)申请人 江苏智云天工科技有限公司 地址 213000 江苏省常州市 常州钟楼经济 开发区玉龙南路280号常州 大数据产 业园4号楼2楼201室 (72)发明人 唐恺　张志琦　赵何　 (74)专利代理 机构 常州佰业腾飞专利代理事务 所(普通合伙) 32231 专利代理师 刘松 (51)Int.Cl. G06T 7/00(2017.01) G06V 10/774(2022.01) G06V 10/82(2022.01) (54)发明名称 工件缺陷检测方法和装置 (57)摘要 本发明提供了一种工件缺陷检测方法和装 置， 其中， 所述方法包括以下步骤： 获取第一数据 集， 其中， 第一数据集为已有的标注缺陷图像； 根 据第一数据集构建基础目标检测模 型； 获取第二 数据集， 其中， 第二数据集为待测工件的未标注 缺陷图像； 根据第一数据集、 第二数据集和基础 目标检测模 型得到目标损失函数； 根据第一数据 集、 第二数据集、 基础目标检测模型和目标损失 函数构建最终目标检测模型； 根据最终目标检测 模型对待测工件进行缺陷检测。 本发 明能够降低 训练数据的获取难度、 以及检测模型的训练成 本， 并且能够自适应待测工件的缺陷检测工作， 此外还具有较高的检测精度。 权利要求书3页 说明书10页 附图1页 CN 115375678 A 2022.11.22 CN 115375678 A 1.一种工件缺陷检测方法， 其特 征在于， 包括以下步骤： 获取第一数据集， 其中， 所述第一数据集 为已有的标注缺陷图像； 根据所述第一数据集构建基础目标检测模型； 获取第二数据集， 其中， 所述第二数据集 为待测工件的未 标注缺陷图像； 根据所述第一数据集、 所述第二数据集和所述基础目标检测模型 得到目标损失函数； 根据所述第一数据集、 所述第二数据集、 所述基础目标检测模型和所述目标损 失函数 构建最终目标检测模型； 根据所述 最终目标检测模型对所述待测工件进行缺陷检测， 所述根据 所述第一数据集、 所述第 二数据集和所述基础目标检测模型得到目标损失函 数， 具体包括以下步骤： 获取所述基础目标检测模 型的特征层的维度和特征矩阵； 获取所述 第一数据集的第一输入图像像素数和所述第二数据集的第二输入图像像素数； 根据所述第 一输入图像像素数、 所述基础目标检测模型的特征层的特征矩阵构建所述第一数据集的第 一特征协方差矩阵； 根据所述第二输入图像像素数、 所述基础目标检测模型 的特征层的特 征矩阵构建所述第二数据集的第二特征协方差矩阵； 根据所述第一特征协方差矩阵、 所述 第二特征协方差矩阵和所述基础目标检测模型的特 征层的维度构建目标损失函数， 其中， 所述第一特 征协方差矩阵的具体表达式为： 其中，CS表示所述第一特征协方差矩阵， 公式下标S表示所述第一数据集， 上标T表示转 置矩阵， 表示所述第一数据集的第一输入图像像素数， 表示所述第一数据集在所 述基础目标检测模型的特 征层对应的特 征矩阵， 1代 表所有值都为1的列向量， 所述第二特 征协方差矩阵的具体表达式为： 其中，CT表示所述第二特征协方差矩阵， 公式下标T表示所述第二数据集， 上标T表示转 置矩阵， 表示所述第二数据集的第二输入图像像素数， 表示所述第二数据集在所 述基础目标检测模型的特 征层对应的特 征矩阵， 1代 表所有值都为1的列向量， 所述目标损失函数的表达式为： 其中， 表示所述目标损失函数， d表示所述基础目标检测模型的特征层的维 度， 表示矩阵范 数的平方。 2.根据权利要求1所述的工件缺陷检测方法， 其特征在于， 所述基础目标检测模型为权　利　要　求　书 1/3 页 2 CN 115375678 A 2YOLO模型。 3.根据权利要求2所述的工件缺陷检测方法， 其特征在于， 所述根据所述第一数据集、 所述第二数据集、 所述基础目标检测模型和所述 目标损失函数构建最终目标检测模型， 具 体包括以下步骤： 将所述目标损失函数设置于所述基础目标检测模型的特征层中， 以构建二 次目标检测 模型； 将所述第一数据集和所述第 二数据集输入至所述二 次目标检测模型中， 以训练得到所 述最终目标检测模型。 4.一种工件缺陷检测装置， 其特 征在于， 包括： 第一获取模块， 所述第一获取模块用于获取第 一数据集， 其中， 所述第 一数据集为已有 的标注缺陷图像； 第一训练模块， 所述第一训练模块用于根据所述第一数据集构建基础目标检测模型； 第二获取模块， 所述第二获取模块用于获取第 二数据集， 其中， 所述第 二数据集为待测 工件的未 标注缺陷图像； 处理模块， 所述处理模块用于根据所述第一数据集、 所述第二数据集和所述基础目标 检测模型 得到目标损失函数； 第二训练模块， 所述第二训练模块用于根据 所述第一数据集、 所述第 二数据集、 所述基 础目标检测模型和所述目标损失函数构建最终目标检测模型； 检测模块， 所述检测模块用于根据 所述最终 目标检测模型对所述待测工件进行缺陷检 测， 所述处理模块具体用于： 获取所述基础目标检测模型的特征层的维度和特征矩阵； 获 取所述第一数据集的第一输入图像像素数和所述第二数据集的第二输入图像像素数； 根据 所述第一输入图像像素数、 所述基础目标检测模型的特征层的特征矩阵构建所述第一数据 集的第一特征协方差矩阵； 根据所述第二输入图像像素数、 所述基础目标检测模型 的特征 层的特征矩阵构建所述第二数据集的第二特征协方差矩阵； 根据所述第一特征协方差矩 阵、 所述第二特征协方差矩阵和所述基础目标检测模型的特征层的维度构建目标损失函 数， 其中， 所述第一特 征协方差矩阵的具体表达式为： 其中，CS表示所述第一特征协方差矩阵， 公式下标S表示所述第一数据集， 上标T表示转 置矩阵， 表示所述第一数据集的第一输入图像像素数， 表示所述第一数据集在所 述基础目标检测模型的特 征层对应的特 征矩阵， 1代 表所有值都为1的列向量， 所述第二特 征协方差矩阵的具体表达式为： 权　利　要　求　书 2/3 页 3 CN 115375678 A 3