[发明专利]一种电缆分类模型训练方法、装置、设备及存储介质

权利要求书

1.一种电缆分类模型训练方法，其特征在于，包括：

采用预设数据库图像对预设分类模型进行预训练；

根据预训练好的预设分类模型，生成初始分类模型；

根据电缆图像样本对所述初始分类模型进行再训练，得到目标分类模型，用于电缆分类。

2.根据权利要求1所述的方法，其特征在于，所述根据预训练好的预设分类模型，生成初始分类模型，包括：

根据电缆分类数量，调整预训练好的预设分类模型的分类层，得到初始分类模型。

3.根据权利要求2所述的方法，其特征在于，所述根据电缆分类数量，调整预训练好的预设分类模型的分类层，得到初始分类模型，包括：

根据电缆分类数量，调整预训练好的预设分类模型的分类层，得到中间分类模型；

在所述中间分类模型的调整后的分类层与前序相邻网络层之间，添加全连接层，得到所述初始分类模型。

4.根据权利要求1-3任一项所述的方法，其特征在于，所述预设分类模型基于Inception V3网络实现。

5.根据权利要求4所述的方法，其特征在于，所述预设分类模型对所述Inception V3网络中第一预设分支模块中的预设子分支进行剔除后得到；和/或，

所述预设分类模型对所述Inception V3网络中第二预设分支模块中的对称子分支进行合并后得到。

6.一种电缆分类模型使用方法，其特征在于，包括：

获取待分类电缆图像；

将所述待分类电缆图像输入至目标分类模型，得到电缆分类结果；其中，所述目标分类模型采用权利要求1-5任一项所述的方法生成。

7.一种电缆分类模型训练装置，其特征在于，包括：

模型预训练模块，用于采用预设数据库图像对预设分类模型进行预训练；

初始分类模型生成模块，用于根据预训练好的预设分类模型，生成初始分类模型；

目标分类模型确定模块，用于根据电缆图像样本对所述初始分类模型进行再训练，得到目标分类模型，用于电缆分类。

8.一种电缆分类模型使用装置，其特征在于，包括：

电缆图像获取模块，用于获取待分类电缆图像；

分类结果确定模块，用于将所述待分类电缆图像输入至目标分类模型，得到电缆分类结果；其中，所述目标分类模型采用权利要求1-5任一项所述的方法生成。

9.一种计算机设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1-5中任一所述的电缆分类模型训练方法，和/或，实现如权利要求6所述的电缆分类模型使用方法。

10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，该程序被处理器执行时实现如权利要求1-5中任一所述的电缆分类模型训练方法，和/或，实现如权利要求6所述的电缆分类模型使用方法。