一种基于深度学习的电池壳表面缺陷检测方法及系统

技术领域

本发明属于电池壳检测技术领域，具体是指一种基于深度学习的电池壳表面缺陷检测方法及系统。

背景技术

电池壳表面缺陷检测是指对电池壳外表面进行检测，以发现和识别可能存在的缺陷和损伤。但是现有的特征选择方法存在受噪声和冗余特征的影响导致准确性低的问题；传统的神经网络模型存在过拟合和损失函数的不平衡的问题；当前的参数搜索算法存在位置更新方法简单导致算法陷入局部最优解，无法找到全局最优解，且计算复杂度高的问题。

发明内容

针对上述情况，为克服现有技术的缺陷，本发明提供了一种基于深度学习的电池壳表面缺陷检测方法及系统，针对现有的特征选择方法存在受噪声和冗余特征的影响导致准确性低的问题，本方案基于皮尔逊相关系数、互信息和加权求和的方法计算相关性值，并将稳定性值和相关性值相结合进行特征选择，提高了特征选择的准确性和可靠性；针对传统的神经网络模型存在过拟合和损失函数的不平衡的问题，本方案基于负样本处理和调整参数优化损失函数，使损失函数更加全面地反映模型的性能，减少模型在训练数据上的过拟合问题，并增强模型的鲁棒性；针对当前的参数搜索算法存在位置更新方法简单导致算法陷入局部最优解，无法找到全局最优解，且计算复杂度高的问题，本方案基于Logistic映射初始化个体位置和基于余弦策略计算能量，扩大搜索范围，并采用柯西突变策略、Logistic映射和精英个人引导机制优化位置更新策略，有效防止算法陷入局部最优解，以协调算法的局部发展和全局搜索能力，增强搜索效率和准确性。

本发明采取的技术方案如下：本发明提供的一种基于深度学习的电池壳表面缺陷检测方法，该方法包括以下步骤：

步骤S1：数据采集，采集电池壳表面图像和对应标签；

步骤S2：特征提取；

步骤S3：特征选择，基于皮尔逊相关系数、互信息和加权求和的方法计算相关性值，并将稳定性值和相关性值相结合进行特征选择；

步骤S4：建立神经网络模型，基于负样本处理和调整参数优化损失函数，实现神经网络模型的建立；

步骤S5：神经网络模型参数搜索，基于Logistic映射初始化个体位置和基于余弦策略计算能量，并采用柯西突变策略、Logistic映射和精英个人引导机制优化位置更新策略，完成参数搜索；

步骤S6：电池壳表面缺陷检测，基于神经网络模型输出的对应标签对电池壳表面进行缺陷检测。

进一步地，在步骤S1中，所述数据采集是采集电池壳表面图像和对应标签，对应标签包括电池壳表面缺陷类型和无缺陷。

进一步地，在步骤S2中，所述特征提取是从采集的数据中提取特征，形成原始特征数据集YS。

进一步地，在步骤S3中，所述特征选择具体包括以下步骤：

步骤S31：初始化，预先设定稳定性阈值δ、第一相关性值阈值ε

步骤S32：计算特征的稳定性值，所用公式如下：

；

式中，A是特征的稳定性值，G是特征，i是特征索引，mode（G

步骤S33：计算第一相关性值，将采集的对应标签作为分类标签，基于皮尔逊相关系数计算特征和分类标签之间的第一相关性值，所用公式如下：

；

式中，B

步骤S34：计算第二相关性值，基于互信息计算特征和分类标签之间的第二相关性值，所用公式如下：

；

式中，C

步骤S35：构建中间特征数据集，将稳定性值小于稳定性阈值δ或稳定性值大于等于稳定性阈值δ且第一相关性值大于等于第一相关性阈值ε

步骤S36：计算相关性值，基于第一相关性值和第二相关性值采用加权求和的方法计算中间特征数据集中特征的相关性值，所用公式如下：

；

式中，E

步骤S37：构建最终特征数据集，将相关性值大于等于相关性阈值ε的特征添加到最终特征数据集TS中。

进一步地，在步骤S4中，所述建立神经网络模型具体包括以下步骤：

步骤S41：初始化神经网络模型，预先设定神经网络模型的层数L和神经元数量；

步骤S42：定义损失函数，引入负样本处理和调整参数优化损失函数，所用公式如下：

；

式中，Loss（W，σ）是损失函数，σ是是网络中的偏置项参数，W是网络中的权重参数，b

进一步地，在步骤S5中，所述神经网络模型参数搜索将神经网络模型的权重W和偏置项σ作为搜索维度，具体包括以下步骤：

步骤S51：初始化，预先设定个体总数量N5、最大迭代次数T、个体维度R、分支参数β、适应度阈值ε

步骤S52：初始化个体位置，采用Logistic映射初始化个体位置，所用公式如下：

D

式中，D

步骤S53：计算最优适应度值和全局最优位置，将基于个体位置的神经网络模型正确率作为对应适应度值M，选择最高的适应度值作为最优适应度值，并将其对应的个体位置作为全局最优位置D

步骤S54：计算能量和步长，基于余弦策略计算能量，所用公式如下：

；

Q=2×（1-rand（0，1））；

式中，P是能量，Q是步长，t是当前迭代次数，cos（）是余弦函数，rand（）是随机数生成函数；

步骤S55：位置更新，若|P|≥1，则转至步骤S551，进入探索阶段；否则，则转至步骤S552，进入局部开采阶段；最后使用精英个人引导机制优化个体位置，步骤如下：

步骤S551：探索阶段，为每个个体随机生成一个（0，1）之间的随机数η，基于η进行探索阶段的位置更新，并采用柯西突变策略对探索阶段位置更新后的全局最优位置进行优化，所用公式如下：

；

；

；

式中，D

步骤S552：局部开采阶段，为每个个体随机生成一个（0，1）之间的随机数γ，若γ≥0.5且|P|≥0.5，则转至步骤S5521，选择第一种位置更新方法进行位置更新；若γ≥0.5且|P|<0.5，则转至步骤S5522，选择第二种位置更新方法进行位置更新；若γ<0.5且|P|≥0.5，则转至步骤S5523，选择第三种位置更新方法进行位置更新；若γ<0.5且|P|<0.5，则转至步骤S5524，选择第四种位置更新方法进行位置更新；最后采用Logistic映射对位置更新后的全局最优位置进行优化，步骤如下：

步骤S5521：第一种位置更新方法，所用公式如下：

△D

D

式中，△D

步骤S5522：第二种位置更新方法，所用公式如下：

D

式中，D

步骤S5523：第三种位置更新方法，所用公式如下：

；

；

YW=D

ZW=YW+U×LF（R）；

；

式中，

步骤S5524：第四种位置更新方法，所用公式如下：

XG=D

HG=XG+P×LF（R）；

；

式中，XG和HG是第四种位置更新方法的两种位置更新策略，

步骤S5525：优化全局最优位置，采用Logistic映射对位置更新后的全局最优位置进行优化，所用公式如下：

；

式中，

步骤S553：优化个体位置，从当前个体中选择前三个适应度值最低的个体位置D

J=2×v×r

；

；

式中，

步骤S56：更新最优适应度值和全局最优位置；

步骤S57：参数确定，当最优适应度值高于适应度阈值ε

进一步地，在步骤S6中，所述电池壳表面缺陷检测是实时采集电池壳表面图像数据，输入至神经网络模型中，基于神经网络模型输出的对应标签对电池壳表面进行缺陷检测。

本发明提供的一种基于深度学习的电池壳表面缺陷检测系统，包括数据采集模块、特征提取模块、特征选择模块、建立神经网络模型模块、神经网络模型参数搜索模块和电池壳表面缺陷检测模块；

所述数据采集模块采集电池壳表面图像和对应标签，对应标签包括电池壳表面缺陷类型和无缺陷，并将采集的数据发送到特征提取模块；

所述特征提取模块从采集的数据中提取特征，形成原始特征数据集，并将原始特征数据集发送至特征选择模块；

所述特征选择模块基于皮尔逊相关系数、互信息和加权求和的方法计算相关性值，并将稳定性值和相关性值相结合进行特征选择，形成最终特征数据集，并将最终特征数据集发送至建立神经网络模型模块；

所述建立神经网络模型模块基于负样本处理和调整参数优化损失函数，实现神经网络模型的建立，并将数据发送至电池壳表面缺陷检测模块；

所述神经网络模型参数搜索模块基于Logistic映射初始化个体位置和基于余弦策略计算能量，并采用柯西突变策略、Logistic映射和精英个人引导机制优化位置更新策略，完成参数搜索，并将数据发送至建立神经网络模型模块；

所述电池壳表面缺陷检测模块实时采集电池壳表面图像数据，输入至神经网络模型中，基于神经网络模型输出的对应标签对电池壳表面进行缺陷检测。

采用上述方案本发明取得的有益效果如下：

（1）针对现有的特征选择方法存在受噪声和冗余特征的影响导致准确性低的问题，本方案基于皮尔逊相关系数、互信息和加权求和的方法计算相关性值，并将稳定性值和相关性值相结合进行特征选择，提高了特征选择的准确性和可靠性。

（2）针对传统的神经网络模型存在过拟合和损失函数的不平衡的问题，本方案基于负样本处理和调整参数优化损失函数，使损失函数更加全面地反映模型的性能，减少模型在训练数据上的过拟合问题，并增强模型的鲁棒性。

（3）针对当前的参数搜索算法存在位置更新方法简单导致算法陷入局部最优解，无法找到全局最优解，且计算复杂度高的问题，本方案基于Logistic映射初始化个体位置和基于余弦策略计算能量，扩大搜索范围，并采用柯西突变策略、Logistic映射和精英个人引导机制优化位置更新策略，有效防止算法陷入局部最优解，以协调算法的局部发展和全局搜索能力，增强搜索效率和准确性。

附图说明

图1为本发明提供的一种基于深度学习的电池壳表面缺陷检测方法的流程示意图；

图2为本发明提供的一种基于深度学习的电池壳表面缺陷检测系统的示意图；

图3为步骤S3的流程示意图；

图4为步骤S4的流程示意图；

图5为步骤S5的流程示意图。

附图用来提供对本发明的进一步理解，并且构成说明书的一部分，与本发明的实施例一起用于解释本发明，并不构成对本发明的限制。

具体实施方式

下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本发明一部分实施例，而不是全部的实施例；基于本发明中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。

在本发明的描述中，需要理解的是，术语“上”、“下”、“前”、“后”、“左”、“右”、“顶”、“底”、“内”、“外”等指示方位或位置关系为基于附图所示的方位或位置关系，仅是为了便于描述本发明和简化描述，而不是指示或暗示所指的装置或元件必须具有特定的方位、以特定的方位构造和操作，因此不能理解为对本发明的限制。

实施例一，参阅图1，本发明提供的一种基于深度学习的电池壳表面缺陷检测方法，该方法包括以下步骤：

步骤S1：数据采集，采集电池壳表面图像和对应标签；

步骤S2：特征提取；

步骤S3：特征选择，基于皮尔逊相关系数、互信息和加权求和的方法计算相关性值，并将稳定性值和相关性值相结合进行特征选择；

步骤S4：建立神经网络模型，基于负样本处理和调整参数优化损失函数，实现神经网络模型的建立；

步骤S5：神经网络模型参数搜索，基于Logistic映射初始化个体位置和基于余弦策略计算能量，并采用柯西突变策略、Logistic映射和精英个人引导机制优化位置更新策略，完成参数搜索；

步骤S6：电池壳表面缺陷检测，基于神经网络模型输出的对应标签对电池壳表面进行缺陷检测。

实施例二，参阅图1，该实施例基于上述实施例，在步骤S1中，数据采集是采集电池壳表面图像和对应标签，对应标签包括电池壳表面缺陷类型和无缺陷。

实施例三，参阅图1，该实施例基于上述实施例，在步骤S2中，特征提取是从采集的数据中提取特征，形成原始特征数据集YS。

实施例四，参阅图1和图3，该实施例基于上述实施例，在步骤S3中，特征选择具体包括以下步骤：

步骤S31：初始化，预先设定稳定性阈值δ、第一相关性值阈值ε

步骤S32：计算特征的稳定性值，所用公式如下：

；

式中，A是特征的稳定性值，G是特征，i是特征索引，mode（G

步骤S33：计算第一相关性值，将采集的对应标签作为分类标签，基于皮尔逊相关系数计算特征和分类标签之间的第一相关性值，所用公式如下：

；

式中，B

步骤S34：计算第二相关性值，基于互信息计算特征和分类标签之间的第二相关性值，所用公式如下：

；

式中，C

步骤S35：构建中间特征数据集，将稳定性值小于稳定性阈值δ或稳定性值大于等于稳定性阈值δ且第一相关性值大于等于第一相关性阈值ε

步骤S36：计算相关性值，基于第一相关性值和第二相关性值采用加权求和的方法计算中间特征数据集中特征的相关性值，所用公式如下：

；

式中，E

步骤S37：构建最终特征数据集，将相关性值大于等于相关性阈值ε的特征添加到最终特征数据集TS中。

通过执行上述操作，针对现有的特征选择方法存在受噪声和冗余特征的影响导致准确性低的问题，本方案基于皮尔逊相关系数、互信息和加权求和的方法计算相关性值，并将稳定性值和相关性值相结合进行特征选择，提高了特征选择的准确性和可靠性。

实施例五，参阅图1和图4，该实施例基于上述实施例，在步骤S4中，建立神经网络模型具体包括以下步骤：

步骤S41：初始化神经网络模型，预先设定神经网络模型的层数L和神经元数量；

步骤S42：定义损失函数，引入负样本处理和调整参数优化损失函数，所用公式如下：

；

式中，Loss（W，σ）是损失函数，σ是是网络中的偏置项参数，W是网络中的权重参数，b

针对传统的神经网络模型存在过拟合和损失函数的不平衡的问题，本方案基于负样本处理和调整参数优化损失函数，使损失函数更加全面地反映模型的性能，减少模型在训练数据上的过拟合问题，并增强模型的鲁棒性。

实施例六，参阅图1和图5，该实施例基于上述实施例，在步骤S5中，神经网络模型参数搜索将神经网络模型的权重W和偏置项σ作为搜索维度，具体包括以下步骤：

步骤S51：初始化，预先设定个体总数量N5、最大迭代次数T、个体维度R、分支参数β、适应度阈值ε

步骤S52：初始化个体位置，采用Logistic映射初始化个体位置，所用公式如下：

D

式中，D

步骤S53：计算最优适应度值和全局最优位置，将基于个体位置的神经网络模型正确率作为对应适应度值M，选择最高的适应度值作为最优适应度值，并将其对应的个体位置作为全局最优位置D

步骤S54：计算能量和步长，基于余弦策略计算能量，所用公式如下：

；

Q=2×（1-rand（0，1））；

式中，P是能量，Q是步长，t是当前迭代次数，cos（）是余弦函数，rand（）是随机数生成函数；

步骤S55：位置更新，若|P|≥1，则转至步骤S551，进入探索阶段；否则，则转至步骤S552，进入局部开采阶段；最后使用精英个人引导机制优化个体位置，步骤如下：

步骤S551：探索阶段，为每个个体随机生成一个（0，1）之间的随机数η，基于η进行探索阶段的位置更新，并采用柯西突变策略对探索阶段位置更新后的全局最优位置进行优化，所用公式如下：

；

；

；

式中，D

步骤S552：局部开采阶段，为每个个体随机生成一个（0，1）之间的随机数γ，若γ≥0.5且|P|≥0.5，则转至步骤S5521，选择第一种位置更新方法进行位置更新；若γ≥0.5且|P|<0.5，则转至步骤S5522，选择第二种位置更新方法进行位置更新；若γ<0.5且|P|≥0.5，则转至步骤S5523，选择第三种位置更新方法进行位置更新；若γ<0.5且|P|<0.5，则转至步骤S5524，选择第四种位置更新方法进行位置更新；最后采用Logistic映射对位置更新后的全局最优位置进行优化，步骤如下：

步骤S5521：第一种位置更新方法，所用公式如下：

△D

D

式中，△D

步骤S5522：第二种位置更新方法，所用公式如下：

D

式中，D

步骤S5523：第三种位置更新方法，所用公式如下：

；

；

YW=D

ZW=YW+U×LF（R）；

；

式中，

步骤S5524：第四种位置更新方法，所用公式如下：

XG=D

HG=XG+P×LF（R）；

；

式中，XG和HG是第四种位置更新方法的两种位置更新策略，

步骤S5525：优化全局最优位置，采用Logistic映射对位置更新后的全局最优位置进行优化，所用公式如下：

；

式中，

步骤S553：优化个体位置，从当前个体中选择前三个适应度值最低的个体位置D

J=2×v×r

；

；

式中，

步骤S56：更新最优适应度值和全局最优位置；

步骤S57：参数确定，当最优适应度值高于适应度阈值ε

通过执行上述操作，针对当前的参数搜索算法存在位置更新方法简单导致算法陷入局部最优解，无法找到全局最优解，且计算复杂度高的问题，本方案基于Logistic映射初始化个体位置和基于余弦策略计算能量，扩大搜索范围，并采用柯西突变策略、Logistic映射和精英个人引导机制优化位置更新策略，有效防止算法陷入局部最优解，以协调算法的局部发展和全局搜索能力，增强搜索效率和准确性。

实施例七，参阅图1，该实施例基于上述实施例，在步骤S6中，电池壳表面缺陷检测是实时采集电池壳表面图像数据，输入至神经网络模型中，基于神经网络模型输出的对应标签对电池壳表面进行缺陷检测。

实施例八，参阅图2，该实施例基于上述实施例，本发明提供的一种基于深度学习的电池壳表面缺陷检测系统，包括数据采集模块、特征提取模块、特征选择模块、建立神经网络模型模块、神经网络模型参数搜索模块和电池壳表面缺陷检测模块；

所述数据采集模块采集电池壳表面图像和对应标签，对应标签包括电池壳表面缺陷类型和无缺陷，并将采集的数据发送到特征提取模块；

所述特征提取模块从采集的数据中提取特征，形成原始特征数据集，并将原始特征数据集发送至特征选择模块；

所述特征选择模块基于皮尔逊相关系数、互信息和加权求和的方法计算相关性值，并将稳定性值和相关性值相结合进行特征选择，形成最终特征数据集，并将最终特征数据集发送至建立神经网络模型模块；

所述建立神经网络模型模块基于负样本处理和调整参数优化损失函数，实现神经网络模型的建立，并将数据发送至电池壳表面缺陷检测模块；

所述神经网络模型参数搜索模块基于Logistic映射初始化个体位置和基于余弦策略计算能量，并采用柯西突变策略、Logistic映射和精英个人引导机制优化位置更新策略，完成参数搜索，并将数据发送至建立神经网络模型模块；

所述电池壳表面缺陷检测模块实时采集电池壳表面图像数据，输入至神经网络模型中，基于神经网络模型输出的对应标签对电池壳表面进行缺陷检测。

需要说明的是，在本文中，诸如第一和第二等之类的关系术语仅仅用来将一个实体或者操作与另一个实体或操作区分开来，而不一定要求或者暗示这些实体或操作之间存在任何这种实际的关系或者顺序。而且，术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含，从而使得包括一系列要素的过程、方法、物品或者设备不仅包括那些要素，而且还包括没有明确列出的其他要素，或者是还包括为这种过程、方法、物品或者设备所固有的要素。

尽管已经示出和描述了本发明的实施例，对于本领域的普通技术人员而言，可以理解在不脱离本发明的原理和精神的情况下可以对这些实施例进行多种变化、修改、替换和变型，本发明的范围由所附权利要求及其等同物限定。

以上对本发明及其实施方式进行了描述，这种描述没有限制性，附图中所示的也只是本发明的实施方式之一，实际的结构并不局限于此。总而言之如果本领域的普通技术人员受其启示，在不脱离本发明创造宗旨的情况下，不经创造性的设计出与该技术方案相似的结构方式及实施例，均应属于本发明的保护范围。