[发明专利]图像分类方法、装置、测试系统及存储介质

说明书

本发明提供一种图像分类方法、装置、测试系统及存储介质。图像分类方法包括：获取至少一待分类图像；设置分类参数；根据所述分类参数，对所述待分类图像进行第一聚类处理，获得第一分类结果；对所述第一分类结果进行特征提取并进行第二聚类处理，获得第二分类结果；显示所述第二分类结果；对所述第二分类结果进行人工复判，获得第三分类结果。

技术领域

本发明是有关于一种图像分类方法，且特别是有关于一种基于人工智能的图像分类方法、装置、测试系统及存储介质。

背景技术

自动光学检测(AOI)系统广泛应用于产品缺陷的自动化检测，随着人工智能(AI)技术的发展，人工智能与自动光学检测的结合大大提高了检测的准确率以及效率。

当前人工智能在进行AI模型训练前期和过程中需要用户先进行图像搜集和分类工作，传统人工分类的方式需要重复多个步骤，特别是首次使用AI应用，会因为大量的人工搜集和分类工作导致用户对AI技术望而却步，搜集并分类300张图像需要2～3天甚至一周的时间，严重影响了AI的应用效果。而且，AI数据的分类繁多复杂，数量达到几十上百种，人工学习周期长，人工分类时间长了也会造成视觉疲劳，导致分类错误，影响AI的训练效果。

因此，如何提供一种能够提高图像分类效率及准确性的图像分类方法，实为需要解决的问题之一。

发明内容

本发明的实施例提供一种图像分类方法、装置、测试系统及存储介质，可以通过自动分类与人工复判相结合的方式，提高图像分类的速度及准确性，提高生产效率。

本发明一实施方式的图像分类方法，包括：获取至少一待分类图像；设置分类参数；根据所述分类参数，对所述待分类图像进行第一聚类处理，获得第一分类结果；对所述第一分类结果进行特征提取并进行第二聚类处理，获得第二分类结果；显示所述第二分类结果；对所述第二分类结果进行人工复判，获得第三分类结果。

上述的方法，其中，所述获得第一分类结果的步骤进一步包括：

将所述待分类图像划分为M组图像，并将所述M组图像分别存储至所对应的存储区域。

上述的方法，其中，所述获得第二分类结果的步骤进一步包括：

将每一所述M组图像，分别划分为N1、N2...Nm类图像，并将每一所述N1、N2...Nm类图像分别存储至所对应的所述M组图像的存储区域。

上述的方法，其中，所述显示所述第二分类结果的步骤进一步包括：

采用图像轮播的方式分别显示每一所述N1、N2...Nm类图像。

上述的方法，其中，所述对所述第二分类结果进行人工复判的步骤进一步包括：

根据所述分类参数，设定快捷键，通过所述快捷键对所述第二分类结果进行人工复判。

上述的方法，其中，所述快捷键可通过输入设备或触控方式进行操作。

上述的方法，其中，所述第一聚类处理或所述第二聚类处理采用无监督聚类算法。

上述的方法，其中，所述方法进一步包括：

根据所述第三分类结果创建学习模型；

根据所述学习模型进行训练并获得分类模型。

上述的方法，其中，所述方法进一步包括：

根据所述分类模型对所述待分类图像进行分类，获得第四分类结果；

显示所述第四分类结果；

对所述第四分类结果进行人工复判，获得第五分类结果。

上述的方法，其中，所述通过所述快捷键对所述第二分类结果进行人工复判的步骤进一步包括：