一种用于图像数据识别模型构建的方法及系统与流程

文档序号:35577266发布日期:2023-09-24 19:33阅读:来源:国知局

技术特征:

1.一种用于图像数据识别模型构建的方法,其特征在于,所述方法包括:

2.根据权利要求1所述的方法,其特征在于,在对所述图像数据集的每个图像数据进行真值标注后,采用gan网络对所述图像数据集进行图像数据扩充。

3.根据权利要求1所述的方法,其特征在于,所述基于所述图像数据集和教师网络模型,采用fsp方法对学生网络各层权重进行初始化,以得到至少两个网络结构相同的学生网络模型包括:

4.根据权利要求3所述的方法,其特征在于,若第一损失不满足预设第一阈值,所述方法还包括:

5.根据权利要求3所述的方法,其特征在于,所述fsp矩阵通过计算相应阶段的输入特征图和输出特征图的内积确定。

6.根据权利要求1所述的方法,其特征在于,所述基于包括真值标注的图像数据集,采用dml蒸馏方法训练每个学生网络模型,当训练后的学生网络模型满足预设条件,将训练后的学生网络模型作为图像数据识别模型包括:

7.根据权利要求6所述的方法,其特征在于,在判断每个学生网络模型的指标之前,所述方法还包括:

8.根据权利要求7所述的方法,其特征在于,每个学生网络模型的kl散度的计算包括:

9.根据权利要求7所述的方法,其特征在于,若训练后的学生网络模型的指标不满足预设条件,所述方法还包括:

10.根据权利要求1所述的方法,其特征在于,所述指标至少包括以下一项:精确率、召回率、准确率、f-score。

11.根据权利要求1所述的方法,其特征在于,所述方法还包括:

12.根据权利要求11所述的方法,其特征在于,所述方法还包括:

13.根据权利要求11所述的方法,其特征在于,所述方法还包括:

14.一种用于图像数据识别模型构建的系统,其特征在于,所述系统包括:

15.一种计算机可读介质,其特征在于,

16.一种用于网络流量数据分类的设备,其特征在于,所述设备包括:


技术总结
本申请提供了一种用于图像数据识别模型构建的方法及系统。其方法包括:获取包括真值标注的图像数据集;基于图像数据集和教师网络模型,采用FSP方法对学生网络各层权重进行初始化,以得到至少两个网络结构相同的学生网络模型;基于包括真值标注的图像数据集,采用DML蒸馏方法训练每个学生网络模型,当训练后的学生网络模型都满足预设条件,将训练后的学生网络模型作为图像数据识别模型。通过该方法,从教师网络模型迁移学习,得到至少两个结构相同且完成各层权重初始化的学生网络模型,可避免数据少无法很好初始化神经网络等问题,且可降低训练难度,采用DML蒸馏方法,得到识别能力有保障的图像数据识别模型,训练周期短、速度快、易部署,适用性广。

技术研发人员:岳腾飞,章曙涵,王夷,张剑,吴翔
受保护的技术使用者:上海芯翌智能科技有限公司
技术研发日:
技术公布日:2024/1/15
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1