目标学生网络模型训练方法及低分辨率图像识别方法

文档序号:34581571发布日期:2023-06-28 14:29阅读:46来源:国知局
目标学生网络模型训练方法及低分辨率图像识别方法

本技术涉及图像识别,尤其涉及一种目标学生网络模型训练方法、低分辨率图像识别方法、装置、设备及存储介质。


背景技术:

1、在过去的几年中,从使用手工特征的传统掌纹识别方法到能够从输入数据中自动学习特征表示的深度学习方法,图像识别方法已经有较大提升。但是,在实际应用中,cnn(convolutional neural networks,卷积神经网络)总是无法在恶劣的条件如低分辨率、低亮度和复杂背景下处理视觉理解任务,且很难容易地装备在边缘计算设备上,如智能手机、嵌入式设备、小型无人机等。这是因为这些方法通常是由复杂的网络架构设计的,并在大规模的高质量的训练图像上学习。然而,边缘计算设备在计算容量和内存使用方面受到限制,它们可能会捕获自然的低分辨率的图像。

2、因此,如何增强对低分辨率图像的识别效果,成为需要解决的问题。

3、在背景技术中公开的上述信息仅用于加强对本技术的背景的理解,因此其可能包含没有形成为本领域普通技术人员所知晓的现有技术的信息。


技术实现思路

1、本技术提供一种目标学生网络模型训练方法、低分辨率图像识别方法、装置、设备及存储介质,用以解决现有技术存在的问题。

2、第一方面,本技术提供一种目标学生网络模型训练方法,包括以下步骤:

3、s11、将高分辨率图像输入至初始学生网络模型和目标协助网络模型,所述初始学生网络模型得到第一中间特征,所述目标协助网络模型得到第二中间特征;

4、s12、将所述第一中间特征和所述第二中间特征输入第一损失函数,得到第一损失值,所述第一损失函数用于使所述初始学生网络模型模拟所述目标协助网络模型的第二中间特征;

5、s13、将低分辨率图像输入至所述初始学生网络模型,通过卷积层中的fsr-w模块得到第三中间特征,所述第三中间特征通过预测损失函数,得到所述初始学生网络模型的第一预测损失值;

6、s14、基于所述第一损失值和所述第一预测损失值,对所述初始学生网络模型进行训练,得到目标学生网络模型。

7、在一些实施例中,所述目标协助网络模型的训练过程,包括以下步骤:

8、s101、将高分辨率图像输入至目标教师网络模型,通过卷积层,得到第四中间特征;

9、s102、将高分辨率图像输入至初始协助网络模型,通过卷积层,得到第五中间特征;

10、s103、计算所述第四中间特征和所述第五中间特征之间的损失,得到特征损失值;

11、s104、所述初始协助网络模型根据所述特征损失值,得到协助预测分数;

12、s105、基于所述目标教师网络模型生成的教师预测分数和所述协助预测分数,计算得到第二预测损失值;

13、s106、根据所述第二预测损失值对所述初始协助网络模型进行回归训练,得到目标协助网络模型。

14、在一些实施例中,所述通过卷积层中的fsr-w模块得到第三中间特征,包括以下步骤:

15、s131、所述低分辨率图像的中间特征进入所述fsr-w模块,所述中间特征包括目标特征图;

16、s132、对所述中间特征进行上采样处理,得到上采样特征图;

17、s133、计算所述目标特征图和所述上采样特征图的残差,得到残差特征图;

18、s134、根据所述目标协助网络模型的所述第二中间特征,得到权重图;

19、s135、将所述上采样特征图、所述残差特征图和所述权重图相加,得到第三中间特征。

20、在一些实施例中,所述目标教师网络模型的训练过程,包括:

21、将高分辨率图像输入初始教师网络模型,对所述初始教师网络模型进行回归训练,得到目标教师网络模型。

22、第二方面,本技术提供一种低分辨率图像识别方法,包括以下步骤:

23、s21、获取待识别图像,所述待识别图像为低分辨率图像;

24、s22、将待识别图像输入至目标学生网络模型中进行识别处理,得到图像识别结果;

25、其中,所述目标学生网络模型通过目标协助网络模型训练得到,且所述目标学生网络模型包括fsr-w模块。

26、在一些实施例中,所述fsr-w模块为高频内容特征超分辨率模块,用于减少由低分辨率图像造成的性能差异。

27、第三方面,本技术提供一种目标学生网络模型训练装置,包括:

28、输入模块,用于将高分辨率图像输入至初始学生网络模型和目标协助网络模型,所述初始学生网络模型得到第一中间特征,所述目标协助网络模型得到第二中间特征;

29、第一处理模块,用于将所述第一中间特征和所述第二中间特征输入第一损失函数,得到第一损失值,所述第一损失函数用于使所述初始学生网络模型模拟所述目标协助网络模型的第二中间特征;

30、第二处理模块,用于将低分辨率图像输入至所述初始学生网络模型,通过卷积层中的fsr-w模块得到第三中间特征,所述第三中间特征通过预测损失函数,得到所述初始学生网络模型的第一预测损失值;

31、训练模块,用于基于所述第一损失值和所述第一预测损失值,对所述初始学生网络模型进行训练,得到目标学生网络模型。

32、第四方面,本技术提供一种低分辨率图像识别装置,包括:

33、获取模块,用于获取待识别图像,所述待识别图像为低分辨率图像;

34、识别模块,用于将待识别图像输入至目标学生网络模型中进行识别处理,得到图像识别结果;其中,所述目标学生网络模型通过目标协助网络模型训练得到,且所述目标学生网络模型包括fsr-w模块。

35、第五方面,本技术提供一种终端设备,包括:

36、存储器,用于存储计算机程序;

37、处理器,用于读取所述存储器中的计算机程序并执行所述的低分辨率图像识别方法对应的操作。

38、第六方面,本技术还提供一种计算机可读存储介质所述计算机可读存储介质中存储有计算机执行指令,所述计算机执行指令被处理器执行时用于实现所述的低分辨率图像识别方法。

39、本技术提供的目标学生网络模型训练方法,包括以下步骤:将高分辨率图像输入至初始学生网络模型和目标协助网络模型,所述初始学生网络模型得到第一中间特征,所述目标协助网络模型得到第二中间特征;将所述第一中间特征和所述第二中间特征输入第一损失函数,得到第一损失值,所述第一损失函数用于使所述初始学生网络模型模拟所述目标协助网络模型的第二中间特征;将低分辨率图像输入至所述初始学生网络模型,通过卷积层中的fsr-w模块得到第三中间特征,所述第三中间特征通过预测损失函数,得到所述初始学生网络模型的第一预测损失值;基于所述第一损失值和所述第一预测损失值,对所述初始学生网络模型进行训练,得到目标学生网络模型。本技术涉及的技术方案,利用目标协助网络模型对初始学生网络模型进行训练,提高初始学生网络模型对高分辨率图像的学习效果,并在初始学生网络模型的卷积层中增加fsr-w模块,使初始学生网络模型能够学习更好的特征表示进行分类,通过上述方式得到目标学生网络模型在低分辨率图像识别方法中能够增强对低分辨率图像的识别效果。

当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1