AI识别石墨大小的方法、装置、计算机设备及存储介质

技术领域

本发明涉及计算机技术领域，尤其涉及AI识别石墨大小的方法、装置、计算机设备及存储介质。

背景技术

在金相组织分析领域，球墨铸铁石墨大小检测是一项极其重要的检测项目，石墨大小级别对产品合格与否的定性和铸铁的力学性能有着直接的影响。目前对于球墨铸铁石墨大小的检测主要有两种方式，一是依靠人工拍照，并将图片与国家标准中的标准图片进行比照，通过肉眼比照定级，二是利用传统测量软件进行手动测量。人工比照法受主观意识影响明显，误差较大，且不同人员标准不统一，而且，传统软件测量法步骤繁琐，耗时长且误差大。目前尚未提出有效的解决方案。

发明内容

为了解决上述技术问题，本发明提出一种AI识别石墨大小的方法、装置、计算机设备及存储介质。本发明的技术方案如下：

一种AI识别石墨大小的方法，包括如下步骤：

获取待测图像，所述待测图像为含有标尺的球墨铸铁图像；

将待测图像输入至预设的第一检测模型中，获得与待测图像相对应的多个识别结果，基于所述多个识别结果，确定所述待测图像中的标尺实际长度；

将待测图像输入至预设的第二检测模型中，识别所述待测图像中的石墨；

基于标尺对应的实际长度，逐个计算石墨的直径和半径，将石墨的最大半径记为A，提取所有直径大于A的石墨，并计算所有直径大于A的石墨的直径平均值；

根据所述直径平均值确定待测图像对应的石墨大小级别。

在一个实施例中，所述将待测图像输入至预设的第一检测模型中，获得与待测图像相对应的多个识别结果，基于所述多个识别结果，确定所述待测图像中的标尺实际长度，具体包括如下步骤，

识别所述待测图像相对应的标尺区域图中的数值区域图和线段区域图，并进行分割；

将分割出的数值区域图输入预设的检测子模型中，获得数值区域图所代表的数值结果；

将待测图像缩放至与数值结果相对应，获得待测图像与标准图之间的缩放比例；

提取分割出的线段区域图中的线段轮廓并计算线段轮廓的像素点长度；

结合缩放比例和线段轮廓的像素点长度，得出对应线段所代表的实际数值，从而得出标尺的实际长度。

在一个实施例中，所述提取分割出的线段区域图中的线段轮廓，包括如下步骤，

将分割出的线段区域图进行边缘检测，得到边缘检测结果；

对边缘结果进行骨架提取，提取出线段轮廓。

在一个实施例中，将所述待测图像进行预处理，获得所述处理后的球墨铸铁图像。

在一个实施例中，所述第一检测模型、所述第二检测模型通过如下步骤训练得到：

将获得待训练的样本数据集，其中，所述样本数据集中的每个样本数据包括：含有标尺的铁素体图像及其对应的标注图像；

对所述样本数据集进行预处理，获得处理后的样本数据集；

使用机器学习算法对所述处理后的样本数据集进行机器学习，生成对应的检测模型。

在一个实施例中，包括如下步骤：

对所述样本数据进行预处理，获得处理后的样本数据。

在一个实施例中，包括如下步骤：

将所述样本数据库中的测试样本集输入检测模型进行验证；

若检测模型的验证准确率未达到预设阈值，则增加样本数据量对检测模型进行再次训练，直至验证准确率达到或超过预设阈值为止。

一种AI识别石墨大小的装置，包括：

获取单元，用于获取待测图像，所述待测图像为含有标尺的球墨铸铁图像；并将待测图像输入至第一识别模型、第二识别模型中；

第一输入单元，用于将待测图像输入至预设的第一检测模型中，获得与待测图像相对应的多个识别结果，基于所述多个识别结果，确定所述待测图像中的标尺实际长度；

第二输入单元，用于将待测图像输入至预设的第二检测模型中，识别所述待测图像中的石墨；

处理单元，用于基于标尺对应的实际长度，逐个计算石墨的直径和半径，将石墨的最大半径记为A，提取所有直径大于A的石墨，并计算所有直径大于A的石墨的直径平均值；用于根据所述直径平均值确定待测图像对应的石墨大小级别。

一种计算机设备，包括存储器和处理器，所述存储器中存储有计算机可读指令，所述计算机可读指令被所述处理器执行时，使得所上述一种AI识别石墨大小的方法的步骤。

一种存储有计算机可读指令的存储介质，所述计算机可读指令被一个或多个处理器执行时，使得一个或多个处理器执行如上述一种AI识别石墨大小的方法的步骤。

基于上述技术方案，本发明的有益效果是：本发明提供一种基于人工智能的方法，通过对已存在的样本数据集进行训练，无需人工干预，对不同级别的球墨铸铁石墨大小图片进行高精度自动识别并定级，最大限度地减少人为主观因素造成的识别错误，使识别、定级更为精准、更快速。

附图说明

下面结合附图对本发明的具体实施方式作进一步详细的说明。

图1：一个实施例中AI识别石墨大小的方法的应用环境图；

图2：一个实施例中AI识别石墨大小的方法流程示意图；

图3：一个实施例中含有石墨的待测图像示意图；

图4：一个实施例中AI识别石墨大小的装置的结构框图；

图5：一个实施例中计算机设备的内部结构图。

具体实施方式

下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述。

本申请实施例提供的一种AI识别石墨大小的方法，可以应用于如图1所示的应用环境中。如图1所示，该应用环境包括计算机设备110。计算机设备110可以获取待测图像，所述待测图像为含有标尺的球墨铸铁图像；计算机设备110可以将待测图像输入至预设的第一检测模型中，获得与待测图像相对应的多个识别结果，基于所述多个识别结果，确定所述待测图像中的标尺实际长度；计算机设备110可以将待测图像输入至预设的第二检测模型中，识别所述待测图像中的石墨；计算机设备110可以基于标尺对应的实际长度，逐个计算石墨的直径和半径，将石墨的最大半径记为A，提取所有直径大于A的石墨，并计算所有直径大于A的石墨的直径平均值；计算机设备110根据所述直径平均值确定待测图像对应的石墨大小级别，其中，计算机设备110可以但不限于是各种个人计算机、笔记本电脑、机器人、平板电脑等。

在一个实施例中，如图2所示，提供一种AI识别石墨大小的方法，包括如下步骤：

步骤202，获取待测图像，所述待测图像为含有标尺的球墨铸铁图像。

本实施例中，可以通过金相显微镜观察大量的金属材料试样的金属组织并采集相应的金相组织图像，这样，就可以得到含有标尺的球墨铸铁图像。

步骤204，将待测图像输入至预设的第一检测模型中，获得与待测图像相对应的多个识别结果，基于所述多个识别结果，确定所述待测图像中的标尺实际长度。

本实施例中，计算机设备可以将待测图像输入至预设的第一检测模型中，获得与待测图像相对应的多个识别结果，计算机设备基于所述多个识别结果，确定所述待测图像中的标尺实际长度，并输出确定结果。其中，计算机设备可以将待测图像进行滤波处理和/或图像增强处理，使得待测图像更清晰,可以提高待测图像的质量。

步骤206，将待测图像输入至预设的第二检测模型中，识别所述待测图像中的石墨。

本实施例中，计算机设备可以将待测图像输入至预设的第二检测模型中，识别所述待测图像中的石墨，并输出识别结果，整个识别过程无需人工操作，降低了人工成本，提高了识别石墨的准确率，其中，计算机设备可以根据待测图像的需求(如:锐化、平滑、降噪等)进行预处理，使得待测图像更清晰,可以提高待测图像的质量。

步骤208，基于标尺对应的实际长度，逐个计算石墨的直径和半径，将石墨的最大半径记为A，提取所有直径大于A的石墨，并计算所有直径大于A的石墨的直径平均值。

步骤210，根据所述直径平均值确定待测图像对应的石墨大小级别。

本实施中，计算机设备根据所述直径平均值与石墨大小级别对应表进行对比，确定待测图像对应的石墨大小级别，计算机设备可以输出确定结果。石墨大小级别对应表如下所示，其中石墨长度指代为所有直径大于A的石墨的直径平均值。

表1石墨大小级别对应表

在一个实施例中，提供一种AI识别石墨大小的方法，包括将待测图像输入至预设的第一检测模型中，获得与待测图像相对应的多个识别结果的过程，过程包括以下步骤：识别所述待测图像相对应的标尺区域图中的数值区域图和线段区域图，并进行分割；将分割出的数值区域图输入预设的检测子模型中，获得数值区域图所代表的数值结果；将待测图像缩放至与数值结果相对应，获得待测图像与标准图之间的缩放比例；提取分割出的线段区域图中的线段轮廓并计算线段轮廓的像素点长度；结合缩放比例和线段轮廓的像素点长度，得出对应线段所代表的实际数值，从而得出标尺的实际长度。

本实施例中，计算机设备可以识别所述待测图像相对应的标尺区域图中的数值区域图和线段区域图，并进行分割；计算机设备将分割出的数值区域图输入预设的检测子模型中，获得数值区域图所代表的数值结果；计算机设备将待测图像缩放至与数值结果相对应，获得待测图像与标准图之间的缩放比例；计算机设备提取分割出的线段区域图中的线段轮廓并计算线段轮廓的像素点长度；计算机设备结合缩放比例和线段轮廓的像素点长度，得出对应线段所代表的实际数值，从而得出标尺的实际长度，计算机设备可以对不同级别、尺寸的球墨铸铁图像进行识别、评级。

在一个实施例中，提供一种AI识别石墨大小的方法，包括提取分割出的线段区域图中的线段轮廓的过程，过程包括如下步骤，将分割出的线段区域图进行边缘检测，得到边缘检测结果；对边缘结果进行骨架提取，提取出线段轮廓。

本实施例中，计算机设备通过微分检测算子Canny对分割出的线段区域图进行边缘检测，计算机设备对边缘检测结果采用K3M算法进行骨架提取，提取出线段轮廓，但不仅限如此，可以采用其他的方式进行边缘检测以及骨架提取。

在一个实施例中，提供一种AI识别石墨大小的方法，将所述待测图像进行预处理，获得所述处理后的球墨铸铁图像。

在一个实施例中，提供一种AI识别石墨大小的方法中第一检测模型、所述第二检测模型通过如下步骤训练得到：

获得样本数据库，所述样本数据库中的每个样本数据包含有标尺的铁素体图像及其对应的标注图像；

使用机器学习算法对所述样本数据库中的训练样本集进行机器学习，生成对应的检测模型。

本实施例中，图3所示，样本数据来源主要是样品拍照、历史检测数据图片和客户图片收集等等，然后上传至计算机设备中保存，形成样本数据库，其样本数据库中的每个样本数据包含有标尺的铁素体图像及其对应的标注图像。将样本数据库中的样本数据按照比例分为训练样本集和测试样本集，计算机设备使用机器学习算法对所述样本数据库中的训练样本集进行机器学习，生成对应的检测模型。基于人工智能的方法，通过对已存在的样本数据集进行训练，无需人工干预，进行自动识别。

在一个实施例中，提供一种AI识别石墨大小的方法，包括优化检测模型的过程，过程包括如下步骤：对所述样本数据进行预处理，获得处理后的样本数据。

在一个实施例中，提供一种AI识别石墨大小的方法，包括优化检测模型的过程，过程包括如下步骤：将所述样本数据库中的测试样本集输入检测模型进行验证；若检测模型的验证准确率未达到预设阈值，则增加样本数据量对检测模型进行再次训练，直至验证准确率达到或超过预设阈值为止。

本实施例中，计算机设备将所述样本数据库中的测试样本集输入检测模型进行验证，若检测模型的验证准确率未达到预设阈值，则增加样本数据量对检测模型进行再次训练，直至验证准确率达到或超过预设阈值为止。当训练样本为500张时，测试准确率可达到98％。

在一个实施例中，如图4所示，一种AI识别石墨大小的装置，包括获取单元310、第一输入单元320、第二输入单元330和处理单元340，其中，

获取单元310，用于获取待测图像，所述待测图像为含有标尺的球墨铸铁图像；并将待测图像输入至第一识别模型、第二识别模型中；

第一输入单元320，用于将待测图像输入至预设的第一检测模型中，获得与待测图像相对应的多个识别结果，基于所述多个识别结果，确定所述待测图像中的标尺实际长度；

第二输入单元330，用于将待测图像输入至预设的第二检测模型中，识别所述待测图像中的石墨；

处理单元340，用于基于标尺对应的实际长度，逐个计算石墨的直径和半径，将石墨的最大半径记为A，提取所有直径大于A的石墨，并计算所有直径大于A的石墨的直径平均值；用于根据所述直径平均值确定待测图像对应的石墨大小级别。

本领域技术人员可以理解，上述实施例阐明的装置、模块或单元，具体可以由计算机芯片或设备实现，或者由具有某种功能的产品来实现，图5中示出的结构，仅仅是与本申请方案相关的部分结构的框图，并不构成对本申请方案所应用于其上的计算机设备的限定，具体的计算机设备可以包括比图中所示更多或更少的部件，或者组合某些部件，或者具有不同的部件布置。

一种计算机设备，包括存储器和处理器，所述存储器中存储有计算机可读指令，所述计算机可读指令被所述处理器执行时，使得该处理器执行计算机程序时实现以下步骤：

获取待测图像，所述待测图像为含有标尺的球墨铸铁图像；

将待测图像输入至预设的第一检测模型中，获得与待测图像相对应的多个识别结果，基于所述多个识别结果，确定所述待测图像中的标尺实际长度；

将待测图像输入至预设的第二检测模型中，识别所述待测图像中的石墨；

基于标尺对应的实际长度，逐个计算石墨的直径和半径，将石墨的最大半径记为A，提取所有直径大于A的石墨，并计算所有直径大于A的石墨的直径平均值；

根据所述直径平均值确定待测图像对应的石墨大小级别。

一种存储有计算机可读指令的存储介质，所述计算机可读指令被一个或多个处理器执行时，使得该处理器执行计算机程序时实现以下步骤：

获取待测图像，所述待测图像为含有标尺的球墨铸铁图像；

将待测图像输入至预设的第一检测模型中，获得与待测图像相对应的多个识别结果，基于所述多个识别结果，确定所述待测图像中的标尺实际长度；

将待测图像输入至预设的第二检测模型中，识别所述待测图像中的石墨；

基于标尺对应的实际长度，逐个计算石墨的直径和半径，将石墨的最大半径记为A，提取所有直径大于A的石墨，并计算所有直径大于A的石墨的直径平均值；

根据所述直径平均值确定待测图像对应的石墨大小级别。

计算机可读介质包括永久性和非永久性、可移动和非可移动媒体可以由任何方法或技术来实现信息存储。信息可以是计算机可读指令、数据结构、程序的模块或其他数据。计算机的存储介质的例子包括，但不限于相变内存(PRAM)、静态随机存取存储器(SRAM)、动态随机存取存储器(DRAM)、其他类型的随机存取存储器(RAM)、只读存储器(ROM)、电可擦除可编程只读存储器(EEPROM)、快闪记忆体或其他内存技术、只读光盘只读存储器(CD-ROM)、数字多功能光盘(DVD)或其他光学存储、磁盒式磁带，磁带磁磁盘存储或其他磁性存储设备或任何其他非传输介质，可用于存储可以被计算设备访问的信息。按照本文中的界定，计算机可读介质不包括暂存电脑可读媒体(transitory media)，如调制的数据信号和载波。

还需要说明的是，术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含，从而使得包括一系列要素的过程、方法、商品或者设备不仅包括那些要素，而且还包括没有明确列出的其他要素，或者是还包括为这种过程、方法、商品或者设备所固有的要素。在没有更多限制的情况下，由语句“包括一个……”限定的要素，并不排除在包括所述要素的过程、方法、商品或者设备中还存在另外的相同要素。

本说明书实施例中的各个实施例均采用递进的方式描述，各个实施例之间相同相似的部分互相参见即可，每个实施例重点说明的都是与其他实施例的不同之处。尤其，对于系统实施例而言，由于其基本相似于方法实施例，所以描述的比较简单，相关之处参见方法实施例的部分说明即可。

以上所述仅为本发明所公开的一种AI识别石墨大小的方法、装置、计算机设备及存储介质的优选实施方式，并非用于限定本说明书实施例的保护范围。凡在本说明书实施例的精神和原则之内，所作的任何修改、等同替换、改进等，均应包含在本说明书实施例的保护范围之内。