图像分类方法、装置及存储介质制造方法及图纸

本申请属于机器视觉技术领域，提供了一种图像分类方法、装置及存储介质，该方法包括：对原始图像进行分割，得到若干待检测图像；对于每一所述待检测图像，将所述待检测图像输入图像分类模型进行计算，得到第一特征数据；对所述待检测图像进行图像特征提取，提取预设目标特征得到第二特征数据；合并所述第一特征数据和所述第二特征数据生成特征图像；通过所述图像分类模型对所述特征图像进行图像分类，输出分类结果。本申请实施例通过将待检测图像提取预设目标特征得到的第二特征数据与卷积得到的第一特征数据合并生成特征图，重新输入至图像分类模型进行图像分类，对图像分类模型的网络结构进行改进，解决分类准确率低的问题。

Image classification method, device and storage medium

【技术实现步骤摘要】
图像分类方法、装置及存储介质
本专利技术涉及机器视觉
，尤其涉及一种图像分类方法、装置及存储介质。
技术介绍
在生产线波峰焊后，由于焊接工艺的不成熟，往往会导致连焊、漏焊等现象，会极大地影响产品的使用寿命、美观等，需要对其进行质量检测。现有技术中通常采用视觉传统算法中的去噪、变换、分割和特征提取的图像处理方法对焊点图像处理，然后通过SVM(SupportVectorMachine)机器对焊点进行分类；或在采用视觉传统算法中的图像处理方法对焊点图像处理后，在焊点数量少时使用SVM对焊点进行分类，在焊点数量多时使用CNN卷积神经网络对焊点进行检测。但现有技术受SVM机器输出结果准确率低的影响，使得采用SVM机器对焊点检测即图像分类时准确率低下。而现有技术采用的另一种焊点检测方法中，虽然在焊点数量多时使用CNN卷积神经网络对焊点进行检测具有较高的准确率，但CNN卷积神经网络算法具有黑盒效应，导致难以准确识别分类。
技术实现思路
有鉴于此，本专利技术实施例提供了一种图像分类方法、装置及存储介质，以解决焊点检测准确率低的本文档来自技高网...

【技术保护点】
1.一种图像分类方法，其特征在于，包括：/n对原始图像进行分割，得到若干待检测图像；/n对于每一所述待检测图像，将所述待检测图像输入图像分类模型进行计算，得到第一特征数据；/n对所述待检测图像进行图像特征提取，提取预设目标特征得到第二特征数据；/n合并所述第一特征数据和所述第二特征数据生成特征图像；/n通过所述图像分类模型对所述特征图像进行图像分类，输出分类结果。/n

【技术特征摘要】
1.一种图像分类方法，其特征在于，包括：
对原始图像进行分割，得到若干待检测图像；
对于每一所述待检测图像，将所述待检测图像输入图像分类模型进行计算，得到第一特征数据；
对所述待检测图像进行图像特征提取，提取预设目标特征得到第二特征数据；
合并所述第一特征数据和所述第二特征数据生成特征图像；
通过所述图像分类模型对所述特征图像进行图像分类，输出分类结果。


2.如权利要求1所述的图像分类方法，其特征在于，在对所述待检测图像进行图像特征提取，提取预设目标特征得到第二特征数据之前，包括：
对所述待检测图像进行灰度化并根据预设图像分割规则将灰度化后的待检测图像均分。


3.如权利要求2所述的图像分类方法，其特征在于，所述预设图像分割规则根据所述第一特征数据的尺寸设定，以使每一均分图像的尺寸与所述第一特征数据的尺寸一致。


4.如权利要求1所述的图像分类方法，其特征在于，在对于每一所述待检测图像，将所述待检测图像输入图像分类模型进行计算，得到第一特征数据之前，还包括：
建立难例池，以根据所述难例池中的易错样本对所述图像分类模型进行迭代训练。


5.如权利要求4所述的图像分类方法，其特征在于，所述方法还包括：
根据训练数据对所述图像分类模型进行图像分类训练；所述训练数据包括正样本和对所述难例池进行过采样得到的样本。


6.如权利要求1-5任一项所述的图像分类方法，其特征在于，所述图像...

【专利技术属性】
技术研发人员：韦奕龙，张强，王朝允，戴盾，
申请(专利权)人：普联技术有限公司，
类型：发明
国别省市：广东;44