[发明专利]一种图像识别方法、终端设备及存储介质

权利要求书

1.一种图像识别方法，其特征在于，包括以下步骤：

S1：采集图像并对其进行标记后组成训练集；

S2：通过训练集对神经网络模型进行训练，并对训练后的神经网络模型进行剪枝，所述对神经网络模型进行剪枝的过程包括以下步骤：

S21：输入神经网络模型中每层卷积层的所有卷积核，设定每层卷积层的最大簇间距，初始化设定层数i＝N，N表示卷积层的总层数；

S22：判断i1是否成立，如果是，进入S26；否则，进入S23；

S23：将第i层卷积层的所有通道进行聚类合并为M个簇，使得合并后的任意两个簇之间的簇间距小于该层对应的最大簇间距；

S24：根据合并后的M个簇，生成第i层卷积层的M个通道，使得每个通道仅保留一个卷积核；

S25：令i＝i-1，返回S22；

S26：输出每层卷积层的所有卷积核；

S3：通过剪枝后的神经网络模型对待识别图像进行识别后输出。

2.根据权利要求1所述的图像识别方法，其特征在于：步骤S23中聚类合并的方法包括以下步骤：

S231：初始化设定每个通道的卷积核作为一个簇；

S232：迭代计算任意两个簇之间的簇间距，当簇间距小于最大簇间距时，将该簇间距对应的两个簇合并为一个簇。

3.根据权利要求2所述的图像识别方法，其特征在于：簇间距的计算方法为根据两个簇中的卷积核之间的距离计算两个簇之间的最小距离、最大距离或平均距离中的一种。

4.根据权利要求3所述的图像识别方法，其特征在于：卷积核之间的距离的计算方法为：将卷积核转换为行向量，通过计算两个行向量之间的距离得到两个卷积核之间的距离。

5.一种图像识别终端设备，其特征在于：包括处理器、存储器以及存储在所述存储器中并在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现如权利要求1～4中任一所述方法的步骤。

6.一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1～4中任一所述方法的步骤。