零部件镂空率确定方法、装置、设备、介质和程序产品与流程

本技术涉及计算机视觉，特别是涉及一种零部件镂空率确定方法、装置、设备、介质和程序产品。

背景技术：

1、在零部件设计、制造及成本评估过程中，镂空率（hollowness rate）是影响零部件重量、结构强度及工艺成本的关键参数。例如，轮辋镂空率是影响轮辋重量和风阻特性的关键参数，在重量预测和成本核算中具有重要意义。然而，相关技术针对不同样式、不同类型的零部件，往往需要重新设计镂空率检测流程或重新采集大量数据进行模型训练，导致操作繁琐且准确性难以保证。

技术实现思路

1、基于此，本技术针对上述技术问题，提供一种能够在准确确定零部件镂空率的前提下简化操作流程的零部件镂空率确定方法、装置、设备、介质和程序产品。

2、第一方面，本技术提供了一种零部件镂空率确定方法，包括：

3、将图像数据和预设的提示词输入已预先完成训练的零部件识别模型，得到零部件识别模型输出的目标边界框；目标边界框标记零部件在图像数据中占据的区域，提示词用于指导零部件识别模型分析推理零部件在图像数据中占据的区域；

4、根据目标边界框，对图像数据中的零部件的图像进行分割，得到零部件的多个候选分割掩码以及每个候选分割掩码的置信度；

5、分别确定每个候选分割掩码对应的零部件的镂空率，得到候选镂空率集合；

6、根据置信度，从候选镂空率集合中确定目标镂空率。

7、上述零部件镂空率确定方法中，采用已预先完成训练的零部件识别模型，基于零部件的图像数据和预设的提示词进行零部件检测以得到目标边界框，使得针对不同零部件只需要调整提示词即可进行零部件检测；通过零部件的图像和目标边界框生成多个候选分割掩码，分别确定每个候选分割掩码对应的零部件的镂空率，并根据每个候选分割掩码的置信度确定目标镂空率，可以针对不同零部件中不同候选分割掩码分别进行镂空率计算，综合不同候选分割掩码对应的镂空率确定目标镂空率，有利于降低因分割掩码形状差异带来的误差，确保了不同零部件镂空率计算的准确性，从而在准确确定不同零部件镂空率的前提下简化了整体操作流程。

8、在第一方面的一可选实施例中，零部件识别模型包括开放词汇检测模型，将图像数据和预设的提示词输入已预先完成训练的零部件识别模型，得到零部件识别模型输出的目标边界框，包括：将图像数据和预设的提示词输入已预先完成训练的开放词汇检测模型，得到开放词汇检测模型输出的多个候选边界框和每个候选边界框的置信度；根据每个候选边界框的置信度，从各候选边界框中确定目标边界框。

9、该可选实施例中，通过开放词汇检测模型基于图像数据和提示词进行零部件检测，并根据多个候选边界框各自的置信度确定目标边界框，能够利用开放词汇检测模型的泛化能力，实现了针对零部件的准确定位，从而有利于提升零部件镂空率确定的准确性。

10、在第一方面的一可选实施例中，对开放词汇检测模型的约束条件包括：边界框置信度阈值，用于约束开放词汇检测模型所输出候选边界框的置信度的取值范围。

11、该可选实施例中，通过边界框置信度阈值针对候选边界框的置信度的取值范围进行约束，以在开放词汇检测模型的零部件检测处理过程中进行有效筛选，可以确保开放词汇检测模型输出的候选边界框的可信度，有利于后续实现零部件的准确定位，并提升零部件镂空率确定的准确性。

12、在第一方面的一可选实施例中，对开放词汇检测模型的约束条件包括：文本匹配度阈值，用于约束开放词汇检测模型所输出候选边界框的文本匹配度的取值范围，文本匹配度用于量化候选边界框所标记的图像区域与提示词之间的关联程度。

13、该可选实施例中，通过文本匹配度阈值针对候选边界框的文本匹配度的取值范围进行约束，以在开放词汇检测模型的零部件检测处理过程中进行有效筛选，可以确保开放词汇检测模型输出的候选边界框的可信度，有利于后续实现零部件的准确定位，并提升零部件镂空率确定的准确性。

14、在第一方面的一可选实施例中，零部件镂空率确定方法还包括：在各候选边界框各自的置信度均小于预设第一置信度阈值的情况下，生成提示信息，以提示图像数据不支持进行镂空率确定。

15、该可选实施例中，在各候选边界框各自的置信度均小于预设第一置信度阈值的情况下，通过提示信息提示图像不支持进行镂空率确定，可以尽可能避免在不可靠的检测结果上继续确定镂空率，有利于降低错误结果传递，从而能够增强镂空率确定的准确性和可靠性。

16、在第一方面的一可选实施例中，根据目标边界框，对图像数据中的零部件的图像进行分割，得到零部件的多个候选分割掩码以及每个候选分割掩码的置信度，包括：从图像数据中提取零部件的多尺度特征；采用预训练的图像分割模型，基于多尺度特征和目标边界框对图像数据中的零部件的图像进行分割，得到多个候选分割掩码以及每个候选分割掩码的置信度。

17、该可选实施例中，通过提取图像数据的多尺度特征，采用图像分割模型结合目标边界框进行分割，以得到针对零部件的多个候选分割掩码以及每个候选分割掩码的置信度，能够利用图像多尺度的特征进行分割，确保分割的准确性与精细度，从而提高了候选分割掩码的准确性。

18、在第一方面的一可选实施例中，分别确定每个候选分割掩码对应的零部件的镂空率，得到候选镂空率集合，包括：针对每个候选分割掩码，基于所针对候选分割掩码的外轮廓确定包络几何形状；根据所针对候选分割掩码和包络几何形状，确定在所针对候选分割掩码中零部件的实体区域与包络几何形状之间的交集面积；确定包络几何形状的参考基准面积，并根据交集面积和参考基准面积，得到所针对候选分割掩码对应的零部件的镂空率；根据各候选分割掩码各自对应的镂空率，得到候选镂空率集合。

19、该可选实施例中，基于候选分割掩码的外轮廓确定包络几何形状，通过候选分割掩码和包络几何形状确定实体区域与包络几何形状之间的交集面积，并结合包络几何形状的参考基准面积确定镂空率，可以基于包络几何形状的参考基准面积建立统一的面积参考，使得各种尺寸和形状的零部件能在同一度量体系下进行镂空率计算，确保了镂空率计算的准确性。

20、在第一方面的一可选实施例中，根据置信度，从候选镂空率集合中确定目标镂空率，包括：根据置信度，从各候选分割掩码中确定目标分割掩码；在目标分割掩码对应的镂空率处于预设的镂空率范围内的情况下，根据目标分割掩码的镂空率，得到零部件的目标镂空率。

21、该可选实施例中，基于候选分割掩码各自的置信度确定目标分割掩码，并基于预设的镂空率范围针对目标分割掩码的镂空率进行合理性校验，在通过合理性校验时根据目标分割掩码的镂空率得到零部件的目标镂空率，从而提升了镂空率检测的抗干扰能力和准确性。

22、在第一方面的一可选实施例中，零部件镂空率确定方法还包括：在目标分割掩码的镂空率不处于镂空率范围内的情况下，确定各剩余分割掩码；剩余分割掩码包括各候选分割掩码中除目标分割掩码外的候选分割掩码；根据各剩余分割掩码各自的置信度和镂空率，从各剩余分割掩码中确定中间分割掩码；中间分割掩码的置信度超过预设的第二置信度阈值、且中间分割掩码的镂空率处于镂空率范围；在中间分割掩码不为空的情况下，基于各中间分割掩码各自的置信度和镂空率，得到零部件的目标镂空率；在中间分割掩码为空的情况下，根据目标分割掩码的镂空率，得到零部件的目标镂空率。

23、该可选实施例中，在目标分割掩码的镂空率未通过合理性校验时，可以从剩余候选掩码中筛选同时满足置信度阈值和镂空率范围的中间分割掩码，以确定零部件的目标镂空率，有利于提升镂空率检测的准确性；而且，在中间分割掩码为空时回退至目标分割掩码得到目标镂空率，确保了镂空率检测处理的稳定性。

24、第二方面，本技术还提供了一种零部件镂空率确定装置，包括：

25、零部件检测模块，用于将图像数据和预设的提示词输入已预先完成训练的零部件识别模型，得到零部件识别模型输出的目标边界框；目标边界框标记零部件在图像数据中占据的区域，提示词用于指导零部件识别模型分析推理零部件在图像数据中占据的区域；

26、分割掩码生成模块，用于根据目标边界框，对图像数据中的零部件的图像进行分割，得到零部件的多个候选分割掩码以及每个候选分割掩码的置信度；

27、镂空率确定模块，用于分别确定每个候选分割掩码对应的零部件的镂空率，得到候选镂空率集合；

28、检测结果获得模块，用于根据置信度，从候选镂空率集合中确定目标镂空率。

29、第三方面，本技术还提供了一种计算机设备，包括存储器和处理器，所述存储器存储有计算机程序，所述处理器执行所述计算机程序时实现上述任一方面的方法的步骤。

30、第四方面，本技术还提供了一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时实现上述任一方面的方法的步骤。

31、第五方面，本技术还提供了一种计算机程序产品，包括计算机程序，该计算机程序被处理器执行时实现上述任一方面的方法的步骤。

32、关于上述第二方面至第五方面中任一技术方案的有益效果，可参照第一方面中的对应技术方案的有益效果，重复之处此处不再列举。